From 908e6559d696c9965e4f91e59a92722e442757b0 Mon Sep 17 00:00:00 2001 From: Francis Couture-Harpin Date: Wed, 2 Jul 2025 23:49:12 -0400 Subject: [PATCH] convert : fix jamba conv1d shape squeezing --- convert_hf_to_gguf.py | 3 +++ 1 file changed, 3 insertions(+) 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)