apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

[BugFix] Fix EB5 model runner compatibility check in worker process (#6673)

Browse Source
This commit is contained in:
sunxin
2026-03-05 19:49:28 +08:00
committed by GitHub
parent 16a393e90e
commit 839bc834eb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/worker/worker_process.py
+2 -2
View File
@@ -563,7 +563,7 @@ class PaddleDisWorkerProc:
if (
not self.parallel_config.use_ep
and not current_platform.is_cuda()
and hasattr(self.worker.model_runner, "not_need_stop")
and not self.worker.model_runner.not_need_stop()
):
self._tp_barrier_wait() if tp_size > 1 else None
@@ -581,7 +581,7 @@ class PaddleDisWorkerProc:
if (
not self.parallel_config.use_ep
and current_platform.is_cuda()
and hasattr(self.worker.model_runner, "current_launch_token_num")
and self.worker.model_runner.current_launch_token_num == 0
):
self._tp_barrier_wait() if tp_size > 1 else None