A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 15, 02:15 UTC
[Misc] Clean up Aimv2 config registration in Ovis config (#20921)
Signed-off-by: Isotr0py <2037008807@qq.com>
Isotr0py committed
Jul 14, 2025 at 23:36 UTC
3fc964433a84bad785d9d0656fd56195462321b8
1 file changed
+5
-5
vllm/transformers_utils/configs/ovis.py
+5
-5
@@ -73,8 +73,6 @@ IMAGE_TOKEN = "<image>"
73
IMAGE_ATOM_ID = -300
74
IMAGE_INDICATOR_IDS = [-301, -302, -303, -304, -305]
75
76
-AutoConfig.register("aimv2", AIMv2Config)
77
-
76
77
# ----------------------------------------------------------------------
78
# Visual Tokenizer Configuration
@@ -105,9 +103,11 @@ class BaseVisualTokenizerConfig(PretrainedConfig):
103
f"expect `backbone_config` to be instance of PretrainedConfig or dict, but got {type(backbone_config)} type"
104
if not isinstance(backbone_config, PretrainedConfig):
105
model_type = backbone_config['model_type']
108
- backbone_config.pop('model_type')
109
- backbone_config = AutoConfig.for_model(model_type,
110
- **backbone_config)
106
+ if model_type != "aimv2":
107
+ backbone_config.pop('model_type')
108
+ backbone_config = AutoConfig.for_model(model_type, **backbone_config)
109
+ else:
110
+ backbone_config = AIMv2Config(**backbone_config)
111
self.backbone_config = backbone_config
112
self.hidden_stride = hidden_stride
113