Skip to content

Commit 91c9ebb

Browse files
[V1] Fix Configs (vllm-project#9971)
1 parent 5459772 commit 91c9ebb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vllm/v1/executor/gpu_executor.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22
from typing import Optional, Tuple
33

4-
from vllm.config import EngineConfig
4+
from vllm.config import VllmConfig
55
from vllm.logger import init_logger
66
from vllm.utils import get_distributed_init_method, get_ip, get_open_port
77
from vllm.v1.outputs import ModelRunnerOutput
@@ -12,7 +12,8 @@
1212

1313
class GPUExecutor:
1414

15-
def __init__(self, vllm_config: EngineConfig) -> None:
15+
def __init__(self, vllm_config: VllmConfig) -> None:
16+
self.vllm_config = vllm_config
1617
self.model_config = vllm_config.model_config
1718
self.cache_config = vllm_config.cache_config
1819
self.lora_config = vllm_config.lora_config

0 commit comments

Comments
 (0)