Fix for lint error complaining of bare except during CI builds.

This commit is contained in:
Clint Herron 2024-04-04 00:33:42 -04:00
parent 5fb1574c81
commit ebae7a96dd

View file

@ -513,7 +513,8 @@ class MPTModel(Model):
def set_vocab(self): def set_vocab(self):
try: try:
self._set_vocab_gpt2() self._set_vocab_gpt2()
except: except Exception:
# Fallback for SEA-LION model
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)