Skip to content

Commit d4e83a0

Browse files
committed
[Bugfix] Fix Dense module loading for sentence-transformers embedding models v12
Signed-off-by: FFFfff1FFFfff <[email protected]>
1 parent 931257d commit d4e83a0

File tree

2 files changed

+12
-11
lines changed

2 files changed

+12
-11
lines changed

requirements/test.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1241,4 +1241,4 @@ yarl==1.17.1
12411241
zipp==3.23.0
12421242
# via importlib-metadata
12431243
zstandard==0.23.0
1244-
# via lm-eval
1244+
# via lm-eval

vllm/transformers_utils/config.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,13 @@
3030
# yapf conflicts with isort for this block
3131
# yapf: disable
3232
from vllm.transformers_utils.configs import (ChatGLMConfig, DeepseekVLV2Config,
33-
EAGLEConfig, JAISConfig,
34-
KimiVLConfig, MedusaConfig,
35-
MLPSpeculatorConfig,
36-
Nemotron_Nano_VL_Config,
37-
NemotronConfig,
38-
OvisConfig, RWConfig,
39-
SpeculatorsConfig,
40-
Step3TextConfig, Step3VLConfig,
41-
UltravoxConfig)
33+
KimiVLConfig, MedusaConfig,
34+
MLPSpeculatorConfig,
35+
Nemotron_Nano_VL_Config,
36+
NemotronConfig, OvisConfig,
37+
RWConfig, SpeculatorsConfig,
38+
Step3TextConfig, Step3VLConfig,
39+
UltravoxConfig)
4240

4341
# yapf: enable
4442
from vllm.transformers_utils.configs.mistral import adapt_config_dict
@@ -936,7 +934,10 @@ def get_hf_file_bytes(file_name: str,
936934

937935
if file_path is None:
938936
try:
939-
hf_hub_file = hf_hub_download(model, file_name, revision=revision, token=_get_hf_token())
937+
hf_hub_file = hf_hub_download(model,
938+
file_name,
939+
revision=revision,
940+
token=_get_hf_token())
940941
file_path = Path(hf_hub_file)
941942
except Exception:
942943
return None

0 commit comments

Comments
 (0)