Skip to content

Commit 51317a8

Browse files
yiyixuxusayakpaul
authored andcommitted
fix _identify_model_variants (#9247)
merge now, will add/fix test next
1 parent de303b9 commit 51317a8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/diffusers/pipelines/pipeline_loading_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -752,12 +752,12 @@ def _fetch_class_library_tuple(module):
752752
def _identify_model_variants(folder: str, variant: str, config: dict) -> dict:
753753
model_variants = {}
754754
if variant is not None:
755-
for folder in os.listdir(folder):
756-
folder_path = os.path.join(folder, folder)
757-
is_folder = os.path.isdir(folder_path) and folder in config
755+
for sub_folder in os.listdir(folder):
756+
folder_path = os.path.join(folder, sub_folder)
757+
is_folder = os.path.isdir(folder_path) and sub_folder in config
758758
variant_exists = is_folder and any(p.split(".")[1].startswith(variant) for p in os.listdir(folder_path))
759759
if variant_exists:
760-
model_variants[folder] = variant
760+
model_variants[sub_folder] = variant
761761
return model_variants
762762

763763

0 commit comments

Comments
 (0)