Skip to content

Commit 999c78f

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

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

vllm/transformers_utils/config.py

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

4141
# yapf: enable
4242
from vllm.transformers_utils.configs.mistral import adapt_config_dict
@@ -929,15 +929,15 @@ def get_hf_file_bytes(file_name: str,
929929
model: Union[str, Path],
930930
revision: Optional[str] = 'main') -> Optional[bytes]:
931931
file_path = try_get_local_file(model=model,
932-
file_name=file_name,
933-
revision=revision)
932+
file_name=file_name,
933+
revision=revision)
934934

935935
if file_path is None:
936936
try:
937937
hf_hub_file = hf_hub_download(model,
938-
file_name,
939-
revision=revision,
940-
token=_get_hf_token())
938+
file_name,
939+
revision=revision,
940+
token=_get_hf_token())
941941
file_path = Path(hf_hub_file)
942942
except Exception:
943943
return None

0 commit comments

Comments
 (0)