We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent de303b9 commit 51317a8Copy full SHA for 51317a8
src/diffusers/pipelines/pipeline_loading_utils.py
@@ -752,12 +752,12 @@ def _fetch_class_library_tuple(module):
752
def _identify_model_variants(folder: str, variant: str, config: dict) -> dict:
753
model_variants = {}
754
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
+ for sub_folder in os.listdir(folder):
+ folder_path = os.path.join(folder, sub_folder)
+ is_folder = os.path.isdir(folder_path) and sub_folder in config
758
variant_exists = is_folder and any(p.split(".")[1].startswith(variant) for p in os.listdir(folder_path))
759
if variant_exists:
760
- model_variants[folder] = variant
+ model_variants[sub_folder] = variant
761
return model_variants
762
763
0 commit comments