[hot-fix] Fix broken CI (#12564)
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user