From af7a26666a743e843ad93eabaabd51b8fc8371dd Mon Sep 17 00:00:00 2001 From: Robin Date: Thu, 5 Sep 2024 17:27:42 -0700 Subject: [PATCH] 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. --- 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 1cdc02e87328..a4f40325ef43 100644 --- a/src/diffusers/utils/hub_utils.py +++ b/src/diffusers/utils/hub_utils.py @@ -497,7 +497,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