Skip to content

Commit a09d403

Browse files
testbotrwightman
authored andcommitted
changed warning to info
1 parent 8470e29 commit a09d403

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/_hub.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)