Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
74056039
Unverified
Commit
74056039
authored
Mar 25, 2026
by
Wei Zhao
Committed by
GitHub
Mar 26, 2026
Browse files
Fix minimax m2.5 nvfp4 kv scales weight loading (#37214)
Signed-off-by:
wzhao18
<
wzhao18.sz@gmail.com
>
parent
d7d51a7e
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
0 deletions
+11
-0
vllm/model_executor/models/minimax_m2.py
vllm/model_executor/models/minimax_m2.py
+11
-0
No files found.
vllm/model_executor/models/minimax_m2.py
View file @
74056039
...
...
@@ -439,6 +439,17 @@ class MiniMaxM2Model(nn.Module):
if
is_pp_missing_parameter
(
name
,
self
):
continue
# Remap qkv_proj.[kv]_scale to attn.[kv]_scale
if
name
.
endswith
((
".k_scale"
,
".v_scale"
)):
remapped_name
=
maybe_remap_kv_scale_name
(
name
,
params_dict
)
if
remapped_name
is
not
None
and
remapped_name
in
params_dict
:
param
=
params_dict
[
remapped_name
]
weight_loader
=
getattr
(
param
,
"weight_loader"
,
default_weight_loader
)
weight_loader
(
param
,
loaded_weight
)
break
param
=
params_dict
[
name
]
weight_loader
=
param
.
weight_loader
weight_loader
(
param
,
loaded_weight
,
shard_id
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment