Update convert_hf_to_gguf.py

Co-authored-by: Sigbjørn Skjæret <[email protected]>

Author: pwilkin

convert_hf_to_gguf.py

@@ -4184,16 +4184,16 @@ def set_vocab(self):
 
 
 @ModelBase.register("Qwen3NextForCausalLM")
-class Qwen3NextModel(Qwen3MoeModel):
+class Qwen3NextModel(Qwen2MoeModel):
     model_arch = gguf.MODEL_ARCH.QWEN3NEXT
 
     def set_gguf_parameters(self):
         super().set_gguf_parameters()
-        self.gguf_writer.add_ssm_conv_kernel(self.find_hparam(["linear_conv_kernel_dim"]))
-        self.gguf_writer.add_ssm_state_size(self.find_hparam(["linear_key_head_dim"]))
-        self.gguf_writer.add_ssm_group_count(self.find_hparam(["linear_num_key_heads"]))
-        self.gguf_writer.add_ssm_time_step_rank(self.find_hparam(["linear_num_value_heads"]))
-        self.gguf_writer.add_ssm_inner_size(self.find_hparam(['linear_value_head_dim']) * self.find_hparam(['linear_num_value_heads']))
+        self.gguf_writer.add_ssm_conv_kernel(self.hparams["linear_conv_kernel_dim"])
+        self.gguf_writer.add_ssm_state_size(self.hparams["linear_key_head_dim"])
+        self.gguf_writer.add_ssm_group_count(self.hparams["linear_num_key_heads"])
+        self.gguf_writer.add_ssm_time_step_rank(self.hparams["linear_num_value_heads"])
+        self.gguf_writer.add_ssm_inner_size(self.hparams["linear_value_head_dim"] * self.hparams["linear_num_value_heads"])
         if (rope_dim := self.hparams.get("head_dim")) is None:
             rope_dim = self.hparams["hidden_size"] // self.hparams["num_attention_heads"]
         self.gguf_writer.add_rope_dimension_count(int(rope_dim * self.hparams.get("partial_rotary_factor", 0.25)))
@@ -4210,7 +4210,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
         elif name.endswith("norm.weight") and not name.endswith("linear_attn.norm.weight"):
             data_torch = data_torch + 1
 
-        yield from Qwen2MoeModel.modify_tensors(self, data_torch, name, bid)
+        yield from super().modify_tensors(self, data_torch, name, bid)
 
 
 @ModelBase.register("RND1")