xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-18 04:45:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Clean up Aimv2 config registration in Ovis config (#20921)

Browse Source 
Signed-off-by: Isotr0py <2037008807@qq.com>
This commit is contained in:
Isotr0py 2025-07-14 23:36:43 +08:00 committed by GitHub
parent 0caf61c08a
commit 3fc964433a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
vllm/transformers_utils/configs/ovis.py
View File

@ -73,8 +73,6 @@ IMAGE_TOKEN = "<image>"
IMAGE_ATOM_ID = -300 IMAGE_ATOM_ID = -300
IMAGE_INDICATOR_IDS = [-301, -302, -303, -304, -305] IMAGE_INDICATOR_IDS = [-301, -302, -303, -304, -305]
AutoConfig.register("aimv2", AIMv2Config)
# ---------------------------------------------------------------------- # ----------------------------------------------------------------------
# Visual Tokenizer Configuration # Visual Tokenizer Configuration
@ -105,9 +103,11 @@ class BaseVisualTokenizerConfig(PretrainedConfig):
f"expect `backbone_config` to be instance of PretrainedConfig or dict, but got {type(backbone_config)} type" f"expect `backbone_config` to be instance of PretrainedConfig or dict, but got {type(backbone_config)} type"
if not isinstance(backbone_config, PretrainedConfig): if not isinstance(backbone_config, PretrainedConfig):
model_type = backbone_config['model_type'] model_type = backbone_config['model_type']
backbone_config.pop('model_type') if model_type != "aimv2":
backbone_config = AutoConfig.for_model(model_type, backbone_config.pop('model_type')
**backbone_config) backbone_config = AutoConfig.for_model(model_type, **backbone_config)
else:
backbone_config = AIMv2Config(**backbone_config)
self.backbone_config = backbone_config self.backbone_config = backbone_config
self.hidden_stride = hidden_stride self.hidden_stride = hidden_stride