Skip to content

GitLab

Unverified Commit edb25693 authored by Liangsheng Yin's avatar Liangsheng Yin Committed by GitHub
Browse files

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

parent 3529c061
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
python/sglang/srt/managers/tp_worker.py
View file @ edb25693
...@@ -100,7 +100,9 @@ class BaseTpWorker(ABC): ...@@ -100,7 +100,9 @@ class BaseTpWorker(ABC):
def update_weights_from_disk(self, recv_req: UpdateWeightFromDiskReqInput): def update_weights_from_disk(self, recv_req: UpdateWeightFromDiskReqInput):
success, message = self.model_runner.update_weights_from_disk( 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 return success, message
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment