Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

download missing models from preset

Browse Source
This commit is contained in:
Manuel Schmid
2023-12-24 13:09:20 +01:00
parent 2e23e2b7b1
commit e72850de70
3 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/config.py
+3 -3
View File
@@ -364,9 +364,9 @@ possible_preset_keys = {
"default_prompt_negative": "Negative Prompt", "default_prompt_negative": "Negative Prompt",
"default_styles": "Styles", "default_styles": "Styles",
"default_aspect_ratio": "Resolution", "default_aspect_ratio": "Resolution",
"checkpoint_downloads": None, "checkpoint_downloads": "checkpoint_downloads",
"embeddings_downloads": None, "embeddings_downloads": "embeddings_downloads",
"lora_downloads": None, "lora_downloads": "lora_downloads",
} }
modules/meta_parser.py
+1 -1
View File
@@ -170,4 +170,4 @@ def parse_meta_from_preset(preset_content):
if settings_key == "default_styles" or settings_key == "default_aspect_ratio": if settings_key == "default_styles" or settings_key == "default_aspect_ratio":
preset_prepared[meta_key] = str(preset_prepared[meta_key]) preset_prepared[meta_key] = str(preset_prepared[meta_key])
return load_parameter_button_click(json.dumps(preset_prepared)) return preset_prepared
webui.py
+10 -2
View File
@@ -16,6 +16,7 @@ import modules.style_sorter as style_sorter
import modules.meta_parser import modules.meta_parser
import args_manager import args_manager
import copy import copy
import launch
from modules.sdxl_styles import legal_style_names from modules.sdxl_styles import legal_style_names
from modules.private_logger import get_current_html_path from modules.private_logger import get_current_html_path
@@ -465,7 +466,14 @@ with shared.gradio_root:
def preset_selection_change(preset): def preset_selection_change(preset):
preset_content = modules.config.try_get_preset_content(preset) if preset != 'initial' else {} preset_content = modules.config.try_get_preset_content(preset) if preset != 'initial' else {}
return modules.meta_parser.parse_meta_from_preset(preset_content) preset_prepared = modules.meta_parser.parse_meta_from_preset(preset_content)
launch.checkpoint_downloads = preset_prepared['checkpoint_downloads']
launch.embeddings_downloads = preset_prepared['embeddings_downloads']
launch.lora_downloads = preset_prepared['lora_downloads']
launch.download_models()
return modules.meta_parser.load_parameter_button_click(json.dumps(preset_prepared))
preset_selection.change(preset_selection_change, inputs=preset_selection, outputs=[ preset_selection.change(preset_selection_change, inputs=preset_selection, outputs=[
advanced_checkbox, advanced_checkbox,
@@ -491,7 +499,7 @@ with shared.gradio_root:
image_seed, image_seed,
generate_button, generate_button,
load_parameter_button load_parameter_button
] + lora_ctrls, queue=False, show_progress=False) \ ] + lora_ctrls, queue=False, show_progress=True) \
.then(fn=style_sorter.sort_styles, inputs=style_selections, outputs=style_selections, queue=False, show_progress=False) \ .then(fn=style_sorter.sort_styles, inputs=style_selections, outputs=style_selections, queue=False, show_progress=False) \
.then(lambda: None, _js='()=>{refresh_style_localization();}') .then(lambda: None, _js='()=>{refresh_style_localization();}')