python fixes

This commit is contained in:
ibrahimkhadraoui
2025-07-04 14:32:11 +04:00
parent 14c37ec047
commit 8bea92261e
2 changed files with 13 additions and 2 deletions

View File

@ -589,6 +589,7 @@ class TensorNameMap:
MODEL_TENSOR.SSM_OUT: (
"model.layers.{bid}.out_proj",
"backbone.layers.{bid}.mixer.out_proj",
"model.layers.{bid}.mamba.out_proj", # falcon-h1
),
MODEL_TENSOR.TIME_MIX_W0: (