Add ovis models support with default buckets

vllm_gaudi/extension/bucketing/vision.py

@@ -11,6 +11,10 @@
     },
 
     # Pixel-based models
+    'ovis': {
+        'is_batch_based': False,
+        'buckets': [1600, 3136, 4096, 6400]
+    },
     'ovis2.5': {
         'is_batch_based': False,
         'buckets': [784, 1600, 3136, 4096, 6400, 7744, 9216, 12544]

vllm_gaudi/models/__init__.py

@@ -15,3 +15,6 @@ def register_model():
     from vllm_gaudi.models.qwen3_vl import HpuQwen3_VLForConditionalGeneration  # noqa: F401
     ModelRegistry.register_model("Qwen3VLForConditionalGeneration",
                                  "vllm_gaudi.models.qwen3_vl:HpuQwen3_VLForConditionalGeneration")
+
+    from vllm_gaudi.models.ovis import HpuOvis  # noqa: F401
+    ModelRegistry.register_model("Ovis", "vllm_gaudi.models.ovis:HpuOvis")

vllm_gaudi/models/ovis.py

@@ -0,0 +1,14 @@
+# SPDX-License-Identifier: Apache-2.0
+
+from vllm.config import VllmConfig
+from vllm.model_executor.models.ovis import (Ovis, OvisMultiModalProcessor, OvisProcessingInfo, OvisDummyInputsBuilder)
+from vllm.multimodal import MULTIMODAL_REGISTRY
+
+
+@MULTIMODAL_REGISTRY.register_processor(OvisMultiModalProcessor,
+                                        info=OvisProcessingInfo,
+                                        dummy_inputs=OvisDummyInputsBuilder)
+class HpuOvis(Ovis):
+
+    def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
+        super().__init__(vllm_config=vllm_config, prefix=prefix)