"vscode:/vscode.git/clone" did not exist on "e152f295020ea2a7ca37be9cabadfaef78464274"
Unverified Commit 1715056f authored by Adam Boeglin's avatar Adam Boeglin Committed by GitHub
Browse files

[Bugfix] Update neuron_executor.py to add optional vision_language_config (#3695)

parent b51c1cc9
from typing import Dict, List, Optional from typing import Dict, List, Optional
from vllm.config import (CacheConfig, DeviceConfig, LoRAConfig, ModelConfig, from vllm.config import (CacheConfig, DeviceConfig, LoRAConfig, ModelConfig,
ParallelConfig, SchedulerConfig) ParallelConfig, SchedulerConfig, VisionLanguageConfig)
from vllm.executor.executor_base import ExecutorBase from vllm.executor.executor_base import ExecutorBase
from vllm.logger import init_logger from vllm.logger import init_logger
from vllm.lora.request import LoRARequest from vllm.lora.request import LoRARequest
...@@ -20,6 +20,7 @@ class NeuronExecutor(ExecutorBase): ...@@ -20,6 +20,7 @@ class NeuronExecutor(ExecutorBase):
scheduler_config: SchedulerConfig, scheduler_config: SchedulerConfig,
device_config: DeviceConfig, device_config: DeviceConfig,
lora_config: Optional[LoRAConfig], lora_config: Optional[LoRAConfig],
vision_language_config: Optional[VisionLanguageConfig],
) -> None: ) -> None:
self.model_config = model_config self.model_config = model_config
self.cache_config = cache_config self.cache_config = cache_config
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment