happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add special handling for default_styles and default_aspect_ratio

This commit is contained in:
Manuel Schmid 2023-12-24 11:25:07 +01:00
parent ccfd4f5094
commit f1bee4b103
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
modules/meta_parser.py
View File

@ -149,13 +149,27 @@ def parse_meta_from_preset(preset_content):
items = preset_content
for settings_key, meta_key in modules.config.possible_preset_keys.items():
if settings_key != "default_loras":
preset_prepared[meta_key] = items[settings_key] if settings_key in items and items[settings_key] is not None else getattr(modules.config, settings_key)
else:
if settings_key == "default_loras":
loras = getattr(modules.config, settings_key)
if settings_key in items:
loras = items[settings_key]
for index, lora in enumerate(loras[:5]):
preset_prepared[f'LoRA {index + 1}'] = ' : '.join(map(str, lora))
elif settings_key == "default_aspect_ratio":
if settings_key in items and items[settings_key] is not None:
default_aspect_ratio = items[settings_key]
width, height = default_aspect_ratio.split('*')
else:
default_aspect_ratio = getattr(modules.config, settings_key)
print(default_aspect_ratio)
width, height = default_aspect_ratio.split('×')
height = height[:height.index(" ")]
print(default_aspect_ratio)
preset_prepared[meta_key] = (width, height)
else:
preset_prepared[meta_key] = items[settings_key] if settings_key in items and items[settings_key] is not None else getattr(modules.config, settings_key)
if settings_key == "default_styles" or settings_key == "default_aspect_ratio":
preset_prepared[meta_key] = str(preset_prepared[meta_key])
return load_parameter_button_click(json.dumps(preset_prepared))