mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-16 12:15:01 +08:00
[Bugfix] Update neuron_executor.py to add optional vision_language_config (#3695)
This commit is contained in:
parent
b51c1cc9d2
commit
1715056fef
@ -1,7 +1,7 @@
|
||||
from typing import Dict, List, Optional
|
||||
|
||||
from vllm.config import (CacheConfig, DeviceConfig, LoRAConfig, ModelConfig,
|
||||
ParallelConfig, SchedulerConfig)
|
||||
ParallelConfig, SchedulerConfig, VisionLanguageConfig)
|
||||
from vllm.executor.executor_base import ExecutorBase
|
||||
from vllm.logger import init_logger
|
||||
from vllm.lora.request import LoRARequest
|
||||
@ -20,6 +20,7 @@ class NeuronExecutor(ExecutorBase):
|
||||
scheduler_config: SchedulerConfig,
|
||||
device_config: DeviceConfig,
|
||||
lora_config: Optional[LoRAConfig],
|
||||
vision_language_config: Optional[VisionLanguageConfig],
|
||||
) -> None:
|
||||
self.model_config = model_config
|
||||
self.cache_config = cache_config
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user