Skip to content

Commit 36e5fe7

Browse files
MODEL_TENSOR.SSM_DT_NORM has defined twice (ggml-org#14991)
* MODEL_TENSOR.SSM_DT_NORM has defined twice, and second overwritten the jamba model's layername * correct order
1 parent 94933c8 commit 36e5fe7

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
@@ -616,6 +616,7 @@ class TensorNameMap:
616616
),
617617

618618
MODEL_TENSOR.SSM_DT_NORM: (
619+
"model.layers.layers.{bid}.mixer.dt_norm.weight", # plamo2
619620
"model.layers.{bid}.mamba.dt_layernorm", # jamba
620621
),
621622

@@ -645,10 +646,6 @@ class TensorNameMap:
645646
"model.layers.layers.{bid}.mixer.D", # plamo2
646647
),
647648

648-
MODEL_TENSOR.SSM_DT_NORM: (
649-
"model.layers.layers.{bid}.mixer.dt_norm.weight", # plamo2
650-
),
651-
652649
MODEL_TENSOR.SSM_NORM: (
653650
"model.layers.{bid}.mamba.norm", # falcon-h1 granite-hybrid
654651
"backbone.layers.{bid}.mixer.norm", # mamba2

0 commit comments

Comments
 (0)