more fixes4

2026-05-14 19:51:56 +00:00
parent bf17bd3fc4
commit d56e86b40e
1 changed files with 4 additions and 1 deletions
--- a/vllm/patches/deepseek_v4.py
+++ b/vllm/patches/deepseek_v4.py
@@ -1323,7 +1323,10 @@ class DeepseekV4Model(nn.Module):
                    continue
                if weight_name not in name:
                    continue
-                name = name.replace(weight_name, param_name)
+                name_mapped = name.replace(weight_name, param_name)
+                if name_mapped not in params_dict:
+                    continue
+                name = name_mapped

                param = params_dict[name]
                weight_loader = param.weight_loader