Skip to content

Commit 272a174

Browse files
correct order
1 parent a567606 commit 272a174

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

gguf-py/gguf/tensor_mapping.py

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

618+
MODEL_TENSOR.SSM_DT_NORM: (
619+
"model.layers.layers.{bid}.mixer.dt_norm.weight", # plamo2
620+
"model.layers.{bid}.mamba.dt_layernorm", # jamba
621+
),
622+
618623
MODEL_TENSOR.SSM_A: (
619624
"model.layers.{bid}.A_log", # mamba-hf
620625
"backbone.layers.{bid}.mixer.A_log", # mamba
@@ -641,11 +646,6 @@ class TensorNameMap:
641646
"model.layers.layers.{bid}.mixer.D", # plamo2
642647
),
643648

644-
MODEL_TENSOR.SSM_DT_NORM: (
645-
"model.layers.layers.{bid}.mixer.dt_norm.weight", # plamo2
646-
"model.layers.{bid}.mamba.dt_layernorm", # jamba
647-
),
648-
649649
MODEL_TENSOR.SSM_NORM: (
650650
"model.layers.{bid}.mamba.norm", # falcon-h1 granite-hybrid
651651
"backbone.layers.{bid}.mixer.norm", # mamba2

0 commit comments

Comments
 (0)