Skip to content

Commit 7a99eab

Browse files
committed
update worker_v1
Signed-off-by: jesse <szxfml@gmail.com>
1 parent 863c441 commit 7a99eab

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm_ascend/worker/worker_v1.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,8 +200,9 @@ def execute_model(
200200

201201
output = self.model_runner.execute_model(scheduler_output,
202202
intermediate_tensors)
203-
if isinstance(output, ModelRunnerOutput):
203+
if isinstance(output, (ModelRunnerOutput, AsyncModelRunnerOutput)):
204204
return output
205+
205206
assert isinstance(output, IntermediateTensors)
206207
parallel_config = self.vllm_config.parallel_config
207208
assert parallel_config.distributed_executor_backend != (

0 commit comments

Comments
 (0)