Skip to content

Commit 44752c8

Browse files
authored
Remove deprecated conversion in Kimi (#47581)
remove it here as well
1 parent aac2512 commit 44752c8

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

src/transformers/conversion_mapping.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,9 @@
2525
GroupWeightRename,
2626
Interleave,
2727
MergeModulelist,
28+
PermuteForRope,
2829
PrefixChange,
2930
Transpose,
30-
VisionUnfuseAndPermuteForRope,
3131
WeightConverter,
3232
WeightRenaming,
3333
WeightTransform,
@@ -471,7 +471,10 @@ def _build_checkpoint_conversion_mapping():
471471
r"attn.k_proj",
472472
r"attn.v_proj",
473473
],
474-
operations=[VisionUnfuseAndPermuteForRope(dim=0, permute_layer_names=["q_proj", "k_proj"])],
474+
operations=[
475+
Chunk(dim=0),
476+
PermuteForRope(subconfig_key="vision_config", permute_layer_names=["q_proj", "k_proj"]),
477+
],
475478
),
476479
],
477480
"deepseek_v4": [

0 commit comments

Comments
 (0)