minor fix

This commit is contained in:
bryan 2024-04-02 16:05:53 +08:00
parent 56e8e63cb0
commit b69e6c0a0f

View file

@ -514,7 +514,7 @@ class MPTModel(Model):
def set_vocab(self): def set_vocab(self):
try: try:
self._set_vocab_gpt2() self._set_vocab_gpt2()
except AttributeError as e: except:
self._set_vocab_sentencepiece() self._set_vocab_sentencepiece()
self.gguf_writer.add_add_bos_token(False) self.gguf_writer.add_add_bos_token(False)
self.gguf_writer.add_pad_token_id(3) self.gguf_writer.add_pad_token_id(3)