Skip to content

Commit a567606

Browse files
MODEL_TENSOR.SSM_DT_NORM has defined twice, and sencond overwritten the jamba model's layername
1 parent 94933c8 commit a567606

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

gguf-py/gguf/tensor_mapping.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -615,10 +615,6 @@ class TensorNameMap:
615615
"model.layers.layers.{bid}.mixer.dt_proj", # plamo2
616616
),
617617

618-
MODEL_TENSOR.SSM_DT_NORM: (
619-
"model.layers.{bid}.mamba.dt_layernorm", # jamba
620-
),
621-
622618
MODEL_TENSOR.SSM_A: (
623619
"model.layers.{bid}.A_log", # mamba-hf
624620
"backbone.layers.{bid}.mixer.A_log", # mamba
@@ -647,6 +643,7 @@ class TensorNameMap:
647643

648644
MODEL_TENSOR.SSM_DT_NORM: (
649645
"model.layers.layers.{bid}.mixer.dt_norm.weight", # plamo2
646+
"model.layers.{bid}.mamba.dt_layernorm", # jamba
650647
),
651648

652649
MODEL_TENSOR.SSM_NORM: (

0 commit comments

Comments
 (0)