Skip to content

Commit 5ba78c7

Browse files
committed
[Bugfix] Fix Dense module loading for sentence-transformers embedding models v5
Signed-off-by: FFFfff1FFFfff <[email protected]>
1 parent eee2905 commit 5ba78c7

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

vllm/transformers_utils/config.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,15 +32,16 @@
3232
from vllm.transformers_utils.configs import (ChatGLMConfig, DeepseekVLV2Config,
3333
EAGLEConfig, JAISConfig,
3434
KimiVLConfig, MedusaConfig,
35-
MllamaConfig, MLPSpeculatorConfig,
35+
MLPSpeculatorConfig,
3636
Nemotron_Nano_VL_Config,
37-
NemotronConfig, NVLM_D_Config,
38-
OvisConfig, RWConfig,
39-
SpeculatorsConfig,
37+
NemotronConfig, OvisConfig,
38+
RWConfig, SpeculatorsConfig,
4039
Step3TextConfig, Step3VLConfig,
4140
UltravoxConfig)
4241
# yapf: enable
4342
from vllm.transformers_utils.configs.mistral import adapt_config_dict
43+
from vllm.transformers_utils.configs.mllama import MllamaConfig
44+
from vllm.transformers_utils.configs.nvlm_d import NVLM_D_Config
4445
from vllm.transformers_utils.utils import check_gguf_file
4546

4647
if envs.VLLM_USE_MODELSCOPE:

0 commit comments

Comments
 (0)