diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 50f029cd3..5c31d9c3d 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -2235,6 +2235,7 @@ class Scheduler( assert flash_cache_success, "Cache flush failed after updating weights" else: logger.error(message) + barrier(group=self.tp_cpu_group) return UpdateWeightsFromTensorReqOutput(success, message) def get_weights_by_name(self, recv_req: GetWeightsByNameReqInput):