Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
71d6de3a
Unverified
Commit
71d6de3a
authored
Jul 04, 2025
by
Isotr0py
Committed by
GitHub
Jul 03, 2025
Browse files
[Misc] Clean up InternVL family config registration (#19992)
Signed-off-by:
Isotr0py
<
2037008807@qq.com
>
parent
536fd330
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
40 additions
and
82 deletions
+40
-82
vllm/transformers_utils/config.py
vllm/transformers_utils/config.py
+22
-6
vllm/transformers_utils/configs/__init__.py
vllm/transformers_utils/configs/__init__.py
+0
-4
vllm/transformers_utils/configs/h2ovl.py
vllm/transformers_utils/configs/h2ovl.py
+0
-16
vllm/transformers_utils/configs/internvl.py
vllm/transformers_utils/configs/internvl.py
+0
-54
vllm/transformers_utils/configs/nvlm_d.py
vllm/transformers_utils/configs/nvlm_d.py
+18
-2
No files found.
vllm/transformers_utils/config.py
View file @
71d6de3a
...
@@ -33,10 +33,8 @@ from vllm.logger import init_logger
...
@@ -33,10 +33,8 @@ from vllm.logger import init_logger
from
vllm.transformers_utils.configs
import
(
ChatGLMConfig
,
Cohere2Config
,
from
vllm.transformers_utils.configs
import
(
ChatGLMConfig
,
Cohere2Config
,
DbrxConfig
,
DeepseekVLV2Config
,
DbrxConfig
,
DeepseekVLV2Config
,
EAGLEConfig
,
ExaoneConfig
,
EAGLEConfig
,
ExaoneConfig
,
H2OVLChatConfig
,
JAISConfig
,
KimiVLConfig
,
InternVLChatConfig
,
JAISConfig
,
MedusaConfig
,
MiniMaxText01Config
,
KimiVLConfig
,
MedusaConfig
,
MiniMaxText01Config
,
MiniMaxVL01Config
,
MllamaConfig
,
MiniMaxVL01Config
,
MllamaConfig
,
MLPSpeculatorConfig
,
MPTConfig
,
MLPSpeculatorConfig
,
MPTConfig
,
NemotronConfig
,
NVLM_D_Config
,
NemotronConfig
,
NVLM_D_Config
,
...
@@ -90,8 +88,6 @@ _CONFIG_REGISTRY: dict[str, type[PretrainedConfig]] = {
...
@@ -90,8 +88,6 @@ _CONFIG_REGISTRY: dict[str, type[PretrainedConfig]] = {
"medusa"
:
MedusaConfig
,
"medusa"
:
MedusaConfig
,
"eagle"
:
EAGLEConfig
,
"eagle"
:
EAGLEConfig
,
"exaone"
:
ExaoneConfig
,
"exaone"
:
ExaoneConfig
,
"h2ovl_chat"
:
H2OVLChatConfig
,
"internvl_chat"
:
InternVLChatConfig
,
"minimax_text_01"
:
MiniMaxText01Config
,
"minimax_text_01"
:
MiniMaxText01Config
,
"minimax_vl_01"
:
MiniMaxVL01Config
,
"minimax_vl_01"
:
MiniMaxVL01Config
,
"nemotron"
:
NemotronConfig
,
"nemotron"
:
NemotronConfig
,
...
@@ -104,6 +100,10 @@ _CONFIG_REGISTRY: dict[str, type[PretrainedConfig]] = {
...
@@ -104,6 +100,10 @@ _CONFIG_REGISTRY: dict[str, type[PretrainedConfig]] = {
**
_CONFIG_REGISTRY_OVERRIDE_HF
**
_CONFIG_REGISTRY_OVERRIDE_HF
}
}
_CONFIG_ATTRS_MAPPING
:
dict
[
str
,
str
]
=
{
"llm_config"
:
"text_config"
,
}
class
ConfigFormat
(
str
,
enum
.
Enum
):
class
ConfigFormat
(
str
,
enum
.
Enum
):
AUTO
=
"auto"
AUTO
=
"auto"
...
@@ -286,6 +286,18 @@ def is_encoder_decoder(config: PretrainedConfig) -> bool:
...
@@ -286,6 +286,18 @@ def is_encoder_decoder(config: PretrainedConfig) -> bool:
return
getattr
(
config
,
"is_encoder_decoder"
,
False
)
return
getattr
(
config
,
"is_encoder_decoder"
,
False
)
def
_maybe_remap_hf_config_attrs
(
config
:
PretrainedConfig
)
->
PretrainedConfig
:
"""Remap config attributes to match the expected names."""
for
old_attr
,
new_attr
in
_CONFIG_ATTRS_MAPPING
.
items
():
if
hasattr
(
config
,
old_attr
):
if
not
hasattr
(
config
,
new_attr
):
config
.
update
({
new_attr
:
getattr
(
config
,
old_attr
)})
delattr
(
config
,
old_attr
)
logger
.
debug
(
"Remapped config attribute '%s' to '%s'"
,
old_attr
,
new_attr
)
return
config
def
get_config
(
def
get_config
(
model
:
Union
[
str
,
Path
],
model
:
Union
[
str
,
Path
],
trust_remote_code
:
bool
,
trust_remote_code
:
bool
,
...
@@ -361,6 +373,9 @@ def get_config(
...
@@ -361,6 +373,9 @@ def get_config(
revision
=
revision
,
revision
=
revision
,
code_revision
=
code_revision
,
code_revision
=
code_revision
,
token
=
_get_hf_token
(),
token
=
_get_hf_token
(),
# some old custom model's config needs
# `has_no_defaults_at_init=True` to work.
has_no_defaults_at_init
=
trust_remote_code
,
**
kwargs
,
**
kwargs
,
)
)
except
ValueError
as
e
:
except
ValueError
as
e
:
...
@@ -376,6 +391,7 @@ def get_config(
...
@@ -376,6 +391,7 @@ def get_config(
raise
RuntimeError
(
err_msg
)
from
e
raise
RuntimeError
(
err_msg
)
from
e
else
:
else
:
raise
e
raise
e
config
=
_maybe_remap_hf_config_attrs
(
config
)
elif
config_format
==
ConfigFormat
.
MISTRAL
:
elif
config_format
==
ConfigFormat
.
MISTRAL
:
config
=
load_params_config
(
model
,
revision
,
**
kwargs
)
config
=
load_params_config
(
model
,
revision
,
**
kwargs
)
...
...
vllm/transformers_utils/configs/__init__.py
View file @
71d6de3a
...
@@ -11,8 +11,6 @@ from vllm.transformers_utils.configs.exaone import ExaoneConfig
...
@@ -11,8 +11,6 @@ from vllm.transformers_utils.configs.exaone import ExaoneConfig
# tiiuae/falcon-7b(-instruct) models. Newer Falcon models will use the
# tiiuae/falcon-7b(-instruct) models. Newer Falcon models will use the
# `FalconConfig` class from the official HuggingFace transformers library.
# `FalconConfig` class from the official HuggingFace transformers library.
from
vllm.transformers_utils.configs.falcon
import
RWConfig
from
vllm.transformers_utils.configs.falcon
import
RWConfig
from
vllm.transformers_utils.configs.h2ovl
import
H2OVLChatConfig
from
vllm.transformers_utils.configs.internvl
import
InternVLChatConfig
from
vllm.transformers_utils.configs.jais
import
JAISConfig
from
vllm.transformers_utils.configs.jais
import
JAISConfig
from
vllm.transformers_utils.configs.kimi_vl
import
KimiVLConfig
from
vllm.transformers_utils.configs.kimi_vl
import
KimiVLConfig
from
vllm.transformers_utils.configs.medusa
import
MedusaConfig
from
vllm.transformers_utils.configs.medusa
import
MedusaConfig
...
@@ -38,8 +36,6 @@ __all__ = [
...
@@ -38,8 +36,6 @@ __all__ = [
"DeepseekVLV2Config"
,
"DeepseekVLV2Config"
,
"MPTConfig"
,
"MPTConfig"
,
"RWConfig"
,
"RWConfig"
,
"H2OVLChatConfig"
,
"InternVLChatConfig"
,
"JAISConfig"
,
"JAISConfig"
,
"MedusaConfig"
,
"MedusaConfig"
,
"EAGLEConfig"
,
"EAGLEConfig"
,
...
...
vllm/transformers_utils/configs/h2ovl.py
deleted
100644 → 0
View file @
536fd330
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
# Adapted from
# https://huggingface.co/h2oai/h2ovl-mississippi-2b/blob/main/configuration_h2ovl_chat.py
# --------------------------------------------------------
# H2OVL-Mississippi
# Copyright (c) 2024 H2O.AI
# Licensed under Apache 2.0 License [see LICENSE for details]
# --------------------------------------------------------
from
.internvl
import
InternVLChatConfig
class
H2OVLChatConfig
(
InternVLChatConfig
):
model_type
=
"h2ovl_chat"
vllm/transformers_utils/configs/internvl.py
deleted
100644 → 0
View file @
536fd330
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
# Adapted from
# https://huggingface.co/OpenGVLab/InternVL2-1B/blob/main/configuration_internvl_chat.py
# --------------------------------------------------------
# InternVL
# Copyright (c) 2024 OpenGVLab
# Licensed under The MIT License [see LICENSE for details]
# --------------------------------------------------------
from
transformers.configuration_utils
import
PretrainedConfig
class
InternVLChatConfig
(
PretrainedConfig
):
model_type
=
'internvl_chat'
is_composition
=
True
def
__init__
(
self
,
vision_config
=
None
,
llm_config
=
None
,
use_backbone_lora
=
0
,
use_llm_lora
=
0
,
select_layer
=-
1
,
force_image_size
=
None
,
downsample_ratio
=
0.5
,
template
=
None
,
dynamic_image_size
=
False
,
use_thumbnail
=
False
,
ps_version
=
'v1'
,
min_dynamic_patch
=
1
,
max_dynamic_patch
=
6
,
**
kwargs
):
super
().
__init__
(
**
kwargs
)
if
vision_config
is
None
:
vision_config
=
{}
if
llm_config
is
None
:
llm_config
=
{}
self
.
vision_config
=
PretrainedConfig
(
**
vision_config
)
self
.
text_config
=
PretrainedConfig
(
**
llm_config
)
self
.
use_backbone_lora
=
use_backbone_lora
self
.
use_llm_lora
=
use_llm_lora
self
.
select_layer
=
select_layer
self
.
force_image_size
=
force_image_size
self
.
downsample_ratio
=
downsample_ratio
self
.
template
=
template
self
.
dynamic_image_size
=
dynamic_image_size
self
.
use_thumbnail
=
use_thumbnail
self
.
ps_version
=
ps_version
# pixel shuffle version
self
.
min_dynamic_patch
=
min_dynamic_patch
self
.
max_dynamic_patch
=
max_dynamic_patch
vllm/transformers_utils/configs/nvlm_d.py
View file @
71d6de3a
...
@@ -8,8 +8,24 @@
...
@@ -8,8 +8,24 @@
# Copyright (c) 2024 NVIDIA
# Copyright (c) 2024 NVIDIA
# Licensed under Apache 2.0 License [see LICENSE for details]
# Licensed under Apache 2.0 License [see LICENSE for details]
# --------------------------------------------------------
# --------------------------------------------------------
from
.internvl
import
InternVLChatConfig
from
transformers
import
Qwen2Config
from
transformers.configuration_utils
import
PretrainedConfig
class
NVLM_D_Config
(
InternVLChat
Config
):
class
NVLM_D_Config
(
Pretrained
Config
):
model_type
=
'NVLM_D'
model_type
=
'NVLM_D'
is_composition
=
True
def
__init__
(
self
,
vision_config
=
None
,
llm_config
=
None
,
**
kwargs
):
super
().
__init__
(
**
kwargs
)
# Handle vision_config initialization
if
vision_config
is
None
:
vision_config
=
{}
# Handle llm_config initialization
if
llm_config
is
None
:
llm_config
=
{}
self
.
vision_config
=
PretrainedConfig
(
**
vision_config
)
self
.
text_config
=
Qwen2Config
(
**
llm_config
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment