Skip to content

Commit 908e655

Browse files
committed
convert : fix jamba conv1d shape squeezing
1 parent 2bcaf64 commit 908e655

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

convert_hf_to_gguf.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5055,6 +5055,9 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
50555055

50565056
new_name = self.map_tensor_name(name)
50575057

5058+
if self.match_model_tensor_name(new_name, gguf.MODEL_TENSOR.SSM_CONV1D, bid):
5059+
data_torch = data_torch.squeeze()
5060+
50585061
if name.endswith(".A_log"):
50595062
logger.debug("A_log --> A ==> " + new_name)
50605063
data_torch = -torch.exp(data_torch)

0 commit comments

Comments
 (0)