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

fix: lint

Browse source
This commit is contained in:
younesbelkada 2024-08-18 11:40:57 +00:00
parent 59a08be755
commit bfa0286866
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

@ -3793,7 +3793,7 @@ class ExaoneModel(Model):
def set_gguf_parameters(self): def set_gguf_parameters(self):
hparams = self.hparams hparams = self.hparams
assert(hparams["activation_function"] == "silu") assert (hparams["activation_function"] == "silu")
max_position_embeddings = hparams["max_position_embeddings"] max_position_embeddings = hparams["max_position_embeddings"]
embed_dim = hparams["hidden_size"] embed_dim = hparams["hidden_size"]
@ -3855,7 +3855,7 @@ class ExaoneModel(Model):
self.gguf_writer.add_tensor(self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FREQS), np.array(rope_factors, dtype=np.float32)) self.gguf_writer.add_tensor(self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FREQS), np.array(rope_factors, dtype=np.float32))
super().prepare_tensors() super().prepare_tensors()
@Model.register("FalconMambaForCausalLM") @Model.register("FalconMambaForCausalLM")
class FalconMambaModel(MambaModel): class FalconMambaModel(MambaModel):