37
37
RWConfig , SpeculatorsConfig ,
38
38
Step3TextConfig , Step3VLConfig ,
39
39
UltravoxConfig )
40
+
40
41
# yapf: enable
41
42
from vllm .transformers_utils .configs .mistral import adapt_config_dict
42
43
from vllm .transformers_utils .utils import check_gguf_file
@@ -65,11 +66,6 @@ def _get_hf_token() -> Optional[str]:
65
66
return token
66
67
return None
67
68
68
-
69
- _CONFIG_REGISTRY_OVERRIDE_HF : dict [str , type [PretrainedConfig ]] = {
70
- "mllama" : MllamaConfig
71
- }
72
-
73
69
_CONFIG_REGISTRY : dict [str , type [PretrainedConfig ]] = {
74
70
"chatglm" : ChatGLMConfig ,
75
71
"deepseek_vl_v2" : DeepseekVLV2Config ,
@@ -83,7 +79,6 @@ def _get_hf_token() -> Optional[str]:
83
79
"eagle" : EAGLEConfig ,
84
80
"speculators" : SpeculatorsConfig ,
85
81
"nemotron" : NemotronConfig ,
86
- "NVLM_D" : NVLM_D_Config ,
87
82
"ovis" : OvisConfig ,
88
83
"ultravox" : UltravoxConfig ,
89
84
"step3_vl" : Step3VLConfig ,
@@ -928,9 +923,9 @@ def get_hf_file_bytes(file_name: str,
928
923
model : Union [str , Path ],
929
924
revision : Optional [str ] = 'main' ) -> Optional [bytes ]:
930
925
file_path = try_get_local_file (model = model ,
931
- file_name = file_name ,
932
- revision = revision )
933
-
926
+ file_name = file_name ,
927
+ revision = revision )
928
+
934
929
if file_path is None :
935
930
try :
936
931
hf_hub_file = hf_hub_download (model ,
@@ -940,12 +935,12 @@ def get_hf_file_bytes(file_name: str,
940
935
file_path = Path (hf_hub_file )
941
936
except Exception :
942
937
return None
943
-
938
+
944
939
if file_path is not None and file_path .is_file ():
945
940
try :
946
941
with open (file_path , 'rb' ) as file :
947
942
return file .read ()
948
943
except Exception :
949
944
return None
950
-
945
+
951
946
return None
0 commit comments