From 27644091191b6f3fb1524109ed70fd23224d761f Mon Sep 17 00:00:00 2001 From: MengqingCao Date: Fri, 30 May 2025 02:57:57 +0000 Subject: [PATCH] [Bugfix][V1] Quick fix on ouput return of model execution Signed-off-by: MengqingCao --- vllm_ascend/worker/worker_v1.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_ascend/worker/worker_v1.py b/vllm_ascend/worker/worker_v1.py index ae6a59eb00..21c67f597e 100644 --- a/vllm_ascend/worker/worker_v1.py +++ b/vllm_ascend/worker/worker_v1.py @@ -173,7 +173,7 @@ def execute_model( scheduler_output: "SchedulerOutput", ) -> Optional[ModelRunnerOutput]: output = self.model_runner.execute_model(scheduler_output) - return output if self.rank == 0 else None + return output if self.is_driver_worker else None def load_model(self) -> None: self.model_runner.load_model()