vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

convert : fix flake8 linter errors

Browse source
This commit is contained in:
Francis Couture-Harpin 2024-02-04 20:41:07 -05:00
parent 766db753c8
commit c52fb3c2de
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
convert-hf-to-gguf.py
View file

@ -1849,9 +1849,9 @@ class MambaModel(Model):
model_arch = gguf.MODEL_ARCH.MAMBA
def set_vocab(self):
vocab_size = self.hparams["vocab_size"];
vocab_size = self.hparams["vocab_size"]
# Round vocab size to next multiple of 8
pad_vocab = self.hparams.get("pad_vocab_size_multiple", 8);
pad_vocab = self.hparams.get("pad_vocab_size_multiple", 8)
self.hparams["vocab_size"] = ((vocab_size + (pad_vocab - 1)) // pad_vocab) * pad_vocab
return self._set_vocab_gpt2()