Skip to content

Commit

Permalink
[Bugfix] Update neuron_executor.py to add optional vision_language_co…
Browse files Browse the repository at this point in the history
  • Loading branch information
adamrb authored and jimpang committed Mar 31, 2024
1 parent 45ebd55 commit 57904a0
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion vllm/executor/neuron_executor.py
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -20,6 +20,7 @@ def __init__(
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
Expand Down

0 comments on commit 57904a0

Please sign in to comment.