From 57904a02975ca8769a5e627b60a8446bd146ed38 Mon Sep 17 00:00:00 2001 From: Adam Boeglin Date: Thu, 28 Mar 2024 10:43:34 -0700 Subject: [PATCH] [Bugfix] Update neuron_executor.py to add optional vision_language_config (#3695) --- vllm/executor/neuron_executor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vllm/executor/neuron_executor.py b/vllm/executor/neuron_executor.py index 9eae7a7df1367..f64c411cc6cb0 100644 --- a/vllm/executor/neuron_executor.py +++ b/vllm/executor/neuron_executor.py @@ -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 @@ 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