Skip to content

Commit 2e25847

Browse files
fix: multi-Step current stream fix
1 parent 7ed7622 commit 2e25847

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/worker/multi_step_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from vllm.model_executor.model_loader.tensorizer import TensorizerConfig
1515
from vllm.sequence import (CompletionSequenceGroupOutput, IntermediateTensors,
1616
Logprob, SequenceGroupMetadata, SequenceOutput)
17-
from vllm.utils import current_stream
17+
from vllm_ascend.utils import current_stream
1818
from vllm.worker.model_runner_base import (
1919
_init_attn_metadata_from_tensor_dict,
2020
_init_frozen_model_input_from_tensor_dict,

0 commit comments

Comments
 (0)