[hot-fix] Fix broken CI (#12564)

2025-11-04 00:03:25 +08:00
parent 3529c061bb
commit edb2569356
1 changed files with 3 additions and 1 deletions
--- a/python/sglang/srt/managers/tp_worker.py
+++ b/python/sglang/srt/managers/tp_worker.py
@@ -100,7 +100,9 @@ class BaseTpWorker(ABC):

    def update_weights_from_disk(self, recv_req: UpdateWeightFromDiskReqInput):
        success, message = self.model_runner.update_weights_from_disk(
-            recv_req.model_path, recv_req.load_format, recv_req.recapture_cuda_graph
+            recv_req.model_path,
+            recv_req.load_format,
+            recapture_cuda_graph=recv_req.recapture_cuda_graph,
        )
        return success, message