File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments