diff --git a/fastdeploy/worker/worker_process.py b/fastdeploy/worker/worker_process.py index 051e5d5933..801c49886d 100644 --- a/fastdeploy/worker/worker_process.py +++ b/fastdeploy/worker/worker_process.py @@ -282,7 +282,8 @@ class PaddleDisWorkerProc: def _broadcast_model_weights_signal(self, src: int, group) -> int: model_weights_signal_tensor = paddle.full(shape=[1], fill_value=self.model_weights_signal[0], dtype="int32") paddle.distributed.broadcast(model_weights_signal_tensor, src=src, group=group) - return model_weights_signal_tensor.item() + value = model_weights_signal_tensor.numpy()[0] + return int(value) def _tp_barrier_wait(self): if current_platform.is_xpu():