Do not trigger autoconversion if local_files_only (#31004)

huggingface · May 24, 2024 · 03935d3 · 03935d3
1 parent 21e259d
commit 03935d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py
@@ -3393,7 +3393,7 @@ def from_pretrained(
                         if resolved_archive_file is not None:
                             is_sharded = True
 
-                    if resolved_archive_file is not None:
+                    if not local_files_only and resolved_archive_file is not None:
                         if filename in [WEIGHTS_NAME, WEIGHTS_INDEX_NAME]:
                             # If the PyTorch file was found, check if there is a safetensors file on the repository
                             # If there is no safetensors file on the repositories, start an auto conversion