|
@@ -2707,7 +2707,7 @@ class XLMRobertaModel(BertModel):
|
|
|
self.gguf_writer.add_token_scores(scores)
|
|
self.gguf_writer.add_token_scores(scores)
|
|
|
self.gguf_writer.add_token_types(toktypes)
|
|
self.gguf_writer.add_token_types(toktypes)
|
|
|
self.gguf_writer.add_add_space_prefix(add_prefix)
|
|
self.gguf_writer.add_add_space_prefix(add_prefix)
|
|
|
- self.gguf_writer.add_token_type_count(1)
|
|
|
|
|
|
|
+ self.gguf_writer.add_token_type_count(self.hparams.get("type_vocab_size", 1))
|
|
|
self.gguf_writer.add_remove_extra_whitespaces(remove_whitespaces)
|
|
self.gguf_writer.add_remove_extra_whitespaces(remove_whitespaces)
|
|
|
if precompiled_charsmap:
|
|
if precompiled_charsmap:
|
|
|
self.gguf_writer.add_precompiled_charsmap(precompiled_charsmap)
|
|
self.gguf_writer.add_precompiled_charsmap(precompiled_charsmap)
|