From bfa0286866885213793e04e3e280013f1fff5f7a Mon Sep 17 00:00:00 2001 From: younesbelkada Date: Sun, 18 Aug 2024 11:40:57 +0000 Subject: [PATCH] fix: lint --- convert_hf_to_gguf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index e7dae3796..aa23bf95c 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -3793,7 +3793,7 @@ class ExaoneModel(Model): def set_gguf_parameters(self): hparams = self.hparams - assert(hparams["activation_function"] == "silu") + assert (hparams["activation_function"] == "silu") max_position_embeddings = hparams["max_position_embeddings"] 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)) super().prepare_tensors() - + @Model.register("FalconMambaForCausalLM") class FalconMambaModel(MambaModel):