flake8 fixes

This commit is contained in:
ibrahimkhadraoui
2025-07-08 13:14:47 +04:00
parent d28c31a90c
commit 9b92648302

View File

@@ -6669,7 +6669,7 @@ class FalconH1Model(Mamba2Model):
# Add any other Falcon Mamba2 specific configuration # Add any other Falcon Mamba2 specific configuration
self.gguf_writer.add_rope_freq_base(self.find_hparam(["rope_theta"])) self.gguf_writer.add_rope_freq_base(self.find_hparam(["rope_theta"]))
@ModelBase.register("HunYuanMoEV1ForCausalLM") @ModelBase.register("HunYuanMoEV1ForCausalLM")
class HunYuanMoEModel(TextModel): class HunYuanMoEModel(TextModel):
model_arch = gguf.MODEL_ARCH.HUNYUAN_MOE model_arch = gguf.MODEL_ARCH.HUNYUAN_MOE