Skip to content

Commit

Permalink
i missed an import
Browse files Browse the repository at this point in the history
  • Loading branch information
kzawora-intel committed Sep 23, 2024
1 parent d1cb336 commit fc3c325
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions vllm/executor/habana_executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,14 +6,16 @@
import os
from typing import Any, Dict, List, Optional, Set, Tuple

from vllm_hpu_extension.profiler import HabanaMemoryProfiler

from vllm.executor.executor_base import ExecutorAsyncBase, ExecutorBase
from vllm.logger import init_logger
from vllm.lora.request import LoRARequest
from vllm.model_executor.layers.sampler import SamplerOutput
from vllm.prompt_adapter.request import PromptAdapterRequest
from vllm.sequence import ExecuteModelRequest
from vllm.utils import (HabanaMemoryProfiler, get_distributed_init_method,
get_ip, get_open_port, make_async)
from vllm.utils import (get_distributed_init_method, get_ip, get_open_port,
make_async)
from vllm.worker.worker_base import WorkerWrapperBase

logger = init_logger(__name__)
Expand Down

0 comments on commit fc3c325

Please sign in to comment.