From 094907eeee6f3569a3b7c756a084b1d8026a616b Mon Sep 17 00:00:00 2001 From: Yuanheng Zhao <54058983+yuanheng-zhao@users.noreply.github.com> Date: Sun, 5 Apr 2026 14:47:08 +0800 Subject: [PATCH] [Chore] Fix Bagel model import compatibility (#2491) Signed-off-by: Yuanheng Zhao --- vllm_omni/model_executor/models/bagel/bagel.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/vllm_omni/model_executor/models/bagel/bagel.py b/vllm_omni/model_executor/models/bagel/bagel.py index 934f434e6..3b4acae51 100644 --- a/vllm_omni/model_executor/models/bagel/bagel.py +++ b/vllm_omni/model_executor/models/bagel/bagel.py @@ -8,7 +8,7 @@ import torch.nn as nn from transformers import BatchFeature from vllm.config import VllmConfig from vllm.config.multimodal import BaseDummyOptions -from vllm.inputs import MultiModalDataDict +from vllm.inputs import ModalityData, MultiModalDataDict from vllm.model_executor.layers.layernorm import RMSNorm as VllmRMSNorm from vllm.model_executor.layers.linear import ( QKVParallelLinear, @@ -27,7 +27,6 @@ from vllm.multimodal.inputs import ( from vllm.multimodal.parse import ( ImageEmbeddingItems, ImageProcessorItems, - ModalityData, ModalityDataItems, MultiModalDataItems, MultiModalDataParser,