diff --git a/vllm/model_executor/models/ovis2_5.py b/vllm/model_executor/models/ovis2_5.py index dd110f7c7cf9..dfbb8c183ae6 100644 --- a/vllm/model_executor/models/ovis2_5.py +++ b/vllm/model_executor/models/ovis2_5.py @@ -10,6 +10,8 @@ from transformers import BaseImageProcessor, BatchFeature, PretrainedConfig from vllm.config import VllmConfig +from vllm.logger import init_logger + from vllm.model_executor.layers.linear import ReplicatedLinear from vllm.model_executor.layers.quantization.base_config import ( QuantizationConfig) @@ -29,9 +31,11 @@ from vllm.multimodal.profiling import BaseDummyInputsBuilder from vllm.sequence import IntermediateTensors from vllm.transformers_utils.processors.ovis2_5 import Ovis2_5Processor - +from vllm.worker.hpu_model_runner import VisionBuckets from .interfaces import MultiModalEmbeddings, SupportsMultiModal +logger = init_logger(__name__) + IMAGE_TOKEN = "" VIDEO_TOKEN = "