[Fix] when run load pretain with local_files_only, local variable 'ca…

…ched_folder' referenced before assignment (huggingface#9376) 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 <[email protected]>
tolgacangoz · Oct 15, 2024 · 79b118e · 79b118e
1 parent 9d06161
commit 79b118e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/utils/hub_utils.py b/src/diffusers/utils/hub_utils.py
@@ -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