Skip to content

Commit 5299829

Browse files
[Bugfix][V1] Fix wrong import in V1 worker (#498)
### What this PR does / why we need it? Fix wrong import in V1 worker. --------- Signed-off-by: shen-shanshan <467638484@qq.com>
1 parent edeadde commit 5299829

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

vllm_ascend/worker/worker_v1.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
from typing import Dict, List, Optional
2222

2323
import torch
24-
import torch.distributed
2524
import torch.nn as nn
2625
import torch_npu
2726
from vllm import envs
@@ -32,12 +31,12 @@
3231
from vllm.logger import init_logger
3332
from vllm.model_executor import set_random_seed
3433
from vllm.platforms import current_platform
35-
from vllm.utils import STR_DTYPE_TO_TORCH_DTYPE
34+
from vllm.utils import STR_DTYPE_TO_TORCH_DTYPE, GiB_bytes
3635
from vllm.v1.core.scheduler import SchedulerOutput
3736
from vllm.v1.kv_cache_interface import (FullAttentionSpec, KVCacheConfig,
3837
KVCacheSpec)
3938
from vllm.v1.outputs import ModelRunnerOutput
40-
from vllm.v1.utils import GiB_bytes, bind_kv_cache
39+
from vllm.v1.utils import bind_kv_cache
4140
from vllm.v1.worker.worker_base import WorkerBase
4241

4342
from vllm_ascend.device_allocator.camem import CaMemAllocator

0 commit comments

Comments
 (0)