From 075ab0d594e38929cdb6c467fa06e8478737242d Mon Sep 17 00:00:00 2001 From: Pierrick HYMBERT Date: Sat, 28 Dec 2024 15:49:47 +0100 Subject: [PATCH] python linter --- convert_hf_to_gguf.py | 1 + 1 file changed, 1 insertion(+) diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index b6adea822d9bc..5a77b18f2e022 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -2564,6 +2564,7 @@ def generate_extra_tensors(self) -> Iterable[tuple[str, Tensor]]: yield (self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FACTORS_LONG), torch.tensor(long_factors, dtype=torch.float32)) yield (self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FACTORS_SHORT), torch.tensor(short_factors, dtype=torch.float32)) + @Model.register("PhiMoEForCausalLM") class PhiMoeModel(Phi3MiniModel): model_arch = gguf.MODEL_ARCH.PHIMOE