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

add LoRA handling

This commit is contained in:
Manuel Schmid 2023-12-23 19:57:56 +01:00
parent f56e3eb3b0
commit 891a1acb62
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/config.py
View File

@ -350,7 +350,7 @@ possible_preset_keys = {
"default_model": "Base Model",
"default_refiner": "Refiner Model",
"default_refiner_switch": "Refiner Switch",
"default_loras": "TODO",
"default_loras": "<processed>",
"default_cfg_scale": "Guidance Scale",
"default_sample_sharpness": "Sharpness",
"default_sampler": "Sampler",

8
modules/meta_parser.py
View File

@ -150,6 +150,12 @@ def parse_meta_from_preset(preset_content):
for k, v in preset_content.items():
if k in modules.config.possible_preset_keys:
if modules.config.possible_preset_keys[k] is not None:
preset_prepared[modules.config.possible_preset_keys[k]] = v
if k != "default_loras":
preset_prepared[modules.config.possible_preset_keys[k]] = v
else:
lora_count = 1
for lora_value in v[:5]:
preset_prepared[f'LoRA {lora_count}'] = ' : '.join(map(str, lora_value))
lora_count += 1
return load_parameter_button_click(json.dumps(preset_prepared))