[Bug] TypeError: maybe_executor_submit() (#13050)

2025-11-11 00:55:13 -08:00
parent 5c0cadd0c6
commit e3b38d7194
1 changed files with 5 additions and 6 deletions
@@ -3723,12 +3723,11 @@ class DeepseekV2ForCausalLM(nn.Module):
                                    param, "weight_loader", default_weight_loader
                                )
                                maybe_executor_submit(
-                                    executor,
+                                    executor=executor,
-                                    futures,
+                                    futures=futures,
-                                    use_async_loading,
+                                    use_async=use_async_loading,
-                                    weight_loader,
+                                    func=weight_loader,
-                                    param,
+                                    func_args=(param, fused_weight),
                                    fused_weight,
                                )
                                cached_wk_and_weights_proj.pop(wk_name)
                                cached_wk_and_weights_proj.pop(weights_proj_name)