diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 0474fd9652..441d6c6d6f 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -5055,6 +5055,9 @@ class JambaModel(TextModel): new_name = self.map_tensor_name(name) + if self.match_model_tensor_name(new_name, gguf.MODEL_TENSOR.SSM_CONV1D, bid): + data_torch = data_torch.squeeze() + if name.endswith(".A_log"): logger.debug("A_log --> A ==> " + new_name) data_torch = -torch.exp(data_torch)