File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -167,14 +167,14 @@ def load_state_dict_from_hf(model_id: str, filename: str = HF_WEIGHTS_NAME):
167167 for safe_filename in _get_safe_alternatives (filename ):
168168 try :
169169 cached_safe_file = hf_hub_download (repo_id = hf_model_id , filename = safe_filename , revision = hf_revision )
170- _logger .warning (f"[{ model_id } ] Safe alternative available for '{ filename } ' (as '{ safe_filename } '). Loading weights using safetensors." )
170+ _logger .info (f"[{ model_id } ] Safe alternative available for '{ filename } ' (as '{ safe_filename } '). Loading weights using safetensors." )
171171 return safetensors .torch .load_file (cached_safe_file , device = "cpu" )
172172 except EntryNotFoundError :
173173 pass
174174
175175 # Otherwise, load using pytorch.load
176176 cached_file = hf_hub_download (hf_model_id , filename = filename , revision = hf_revision )
177- _logger .warning (f"[{ model_id } ] Safe alternative not found for '{ filename } '. Loading weights using default pytorch." )
177+ _logger .info (f"[{ model_id } ] Safe alternative not found for '{ filename } '. Loading weights using default pytorch." )
178178 return torch .load (cached_file , map_location = 'cpu' )
179179
180180
You can’t perform that action at this time.
0 commit comments