AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

fix _identify_model_variants (#9247)

Browse Source 
merge now, will add/fix test next
This commit is contained in:
YiYi Xu
2024-08-22 12:00:17 -10:00
committed by GitHub
parent 805bf33fa7
commit dc07fc29da
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/diffusers/pipelines/pipeline_loading_utils.py
View File

@@ -752,12 +752,12 @@ def _fetch_class_library_tuple(module):
def _identify_model_variants(folder: str, variant: str, config: dict) -> dict:
model_variants = {}
if variant is not None:
for folder in os.listdir(folder):
folder_path = os.path.join(folder, folder)
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
variant_exists = is_folder and any(p.split(".")[1].startswith(variant) for p in os.listdir(folder_path))
if variant_exists:
model_variants[folder] = variant
model_variants[sub_folder] = variant
return model_variants