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

RobinXL · yiyixuxu · web-flow · commit 79b118e863c2 · 2024-10-14T20:49:36.000-10:00
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 &lt;yixu310@gmail.com&gt;
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
 

Original file line number	Diff line number	Diff line change
`@@ -496,7 +496,7 @@ def _get_checkpoint_shard_files(`
`496`	`496`	`local_dir=cache_dir, subfolder=subfolder, original_shard_filenames=original_shard_filenames`
`497`	`497`	`)`
`498`	`498`	`if subfolder is not None:`
`499`		`- cached_folder = os.path.join(cached_folder, subfolder)`
	`499`	`+ cached_folder = os.path.join(cache_dir, subfolder)`
`500`	`500`
`501`	`501`	`return cached_folder, sharded_metadata`
`502`	`502`