convert : match ssm_conv tensors by type

Author: compilade

convert_hf_to_gguf.py

@@ -3803,8 +3803,6 @@ def set_gguf_parameters(self):
     _tok_embd = None
 
     def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iterable[tuple[str, Tensor]]:
-        del bid  # unused
-
         output_name = self.format_tensor_name(gguf.MODEL_TENSOR.OUTPUT)
         tok_embd_name = self.format_tensor_name(gguf.MODEL_TENSOR.TOKEN_EMBD)
 
@@ -3815,7 +3813,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
             data_torch = -torch.exp(data_torch)
 
         # [4 1 8192 1] -> [4 8192 1 1]
-        if new_name.endswith(".ssm_conv1d"):
+        if self.match_model_tensor_name(new_name, gguf.MODEL_TENSOR.SSM_CONV1D, bid):
             data_torch = data_torch.squeeze()
 
         # assuming token_embd.weight is seen before output.weight