From 79b118e863c2caf593a8fb40b72fe2e67527f22a Mon Sep 17 00:00:00 2001 From: Robin Date: Mon, 14 Oct 2024 23:49:36 -0700 Subject: [PATCH] [Fix] when run load pretain with local_files_only, local variable 'cached_folder' referenced before assignment (#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 --- src/diffusers/utils/hub_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/utils/hub_utils.py b/src/diffusers/utils/hub_utils.py index a79c6cdbfe..3ff859f17f 100644 --- 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