diff --git a/gguf-py/gguf/tensor_mapping.py b/gguf-py/gguf/tensor_mapping.py index 15adbfa78..df490fc80 100644 --- a/gguf-py/gguf/tensor_mapping.py +++ b/gguf-py/gguf/tensor_mapping.py @@ -616,6 +616,7 @@ class TensorNameMap: ), MODEL_TENSOR.SSM_DT_NORM: ( + "model.layers.layers.{bid}.mixer.dt_norm.weight", # plamo2 "model.layers.{bid}.mamba.dt_layernorm", # jamba ), @@ -645,10 +646,6 @@ class TensorNameMap: "model.layers.layers.{bid}.mixer.D", # plamo2 ), - MODEL_TENSOR.SSM_DT_NORM: ( - "model.layers.layers.{bid}.mixer.dt_norm.weight", # plamo2 - ), - MODEL_TENSOR.SSM_NORM: ( "model.layers.{bid}.mamba.norm", # falcon-h1 granite-hybrid "backbone.layers.{bid}.mixer.norm", # mamba2