remove todo

This commit is contained in:
younesbelkada
2025-07-08 10:44:59 +04:00
parent 2dee7cf964
commit a846d02327
2 changed files with 2 additions and 6 deletions

View File

@@ -583,6 +583,7 @@ class TensorNameMap:
),
MODEL_TENSOR.SSM_NORM: (
"model.layers.{bid}.mamba.norm", # falcon-h1
"backbone.layers.{bid}.mixer.norm", # mamba2
),
@@ -1177,10 +1178,6 @@ class TensorNameMap:
"resampler.attn.out_proj",
),
MODEL_TENSOR.SSM_NORM: (
"model.layers.{bid}.mamba.norm",
),
MODEL_TENSOR.V_RESMPL_KV: (
"resampler.kv_proj",
),