AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Fix] when run load pretain with local_files_only, local variable 'cached_folder' referenced before assignment (#9376)

Browse Source 
Fix local variable 'cached_folder' referenced before assignment in hub_utils.py

Fix when use `local_files_only=True` with `subfolder`, local variable 'cached_folder' referenced before assignment issue.

Co-authored-by: YiYi Xu <yixu310@gmail.com>
This commit is contained in:
Robin
2024-10-14 23:49:36 -07:00
committed by GitHub
parent 9d0616189e
commit 79b118e863
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/utils/hub_utils.py
View File

@@ -496,7 +496,7 @@ def _get_checkpoint_shard_files(
local_dir=cache_dir, subfolder=subfolder, original_shard_filenames=original_shard_filenames
)
if subfolder is not None:
cached_folder = os.path.join(cached_folder, subfolder)
cached_folder = os.path.join(cache_dir, subfolder)
return cached_folder, sharded_metadata