riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Add warning when failed to acquire other user's lock at model download (#37395)

Browse Source
This commit is contained in:
Manuel de Prada Corral 2025-04-10 11:18:27 +02:00 committed by GitHub
parent 9cda4265d6
commit bc0150bb04
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/transformers/utils/hub.py
View File

@ -465,6 +465,12 @@ def cached_files(
"for this model name. Check the model page at "
f"'https://huggingface.co/{path_or_repo_id}' for available revisions."
) from e
elif isinstance(e, PermissionError):
raise OSError(
f"PermissionError at {e.filename} when downloading {path_or_repo_id}. "
"Check cache directory permissions. Common causes: 1) another user is downloading the same model (please wait); "
"2) a previous download was canceled and the lock file needs manual removal."
) from e
# Now we try to recover if we can find all files correctly in the cache
resolved_files = [
@ -870,7 +876,7 @@ class PushToHubMixin:
)
if os.path.isdir(repo_path_or_name):
# repo_path: infer repo_id from the path
repo_id = repo_id.split(os.path.sep)[-1]
repo_id = repo_path_or_name.split(os.path.sep)[-1]
working_dir = repo_id
else:
# repo_name: use it as repo_id