19 lines
917 B
Diff
19 lines
917 B
Diff
--- a/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4_moe.py
|
|
+++ b/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4_moe.py
|
|
@@ -537,6 +537,7 @@
|
|
if is_pp_missing_parameter(name, self):
|
|
continue
|
|
|
|
+ if ('k_scale' in name or 'v_scale' in name) and name not in params_dict: continue
|
|
param = params_dict[name]
|
|
weight_loader = param.weight_loader
|
|
weight_loader(param, loaded_weight, shard_id)
|
|
@@ -596,6 +597,7 @@
|
|
if is_pp_missing_parameter(name, self):
|
|
continue
|
|
|
|
+ if ('k_scale' in name or 'v_scale' in name) and name not in params_dict: continue
|
|
param = params_dict[name]
|
|
weight_loader = getattr(
|
|
param, "weight_loader", default_weight_loader
|