mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: use minimum image number of preset and config to prevent UI overflow

Browse Source
This commit is contained in:
Manuel Schmid 2024-03-23 12:44:21 +01:00
parent 04ac2c6d17
commit a9b6e5ccf7
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/meta_parser.py
View File

@ -29,7 +29,7 @@ def load_parameter_button_click(raw_metadata: dict | str, is_generating: bool):
results = [len(loaded_parameter_dict) > 0]
get_int('image_number', 'Image Number', loaded_parameter_dict, results)
get_image_number('image_number', 'Image Number', loaded_parameter_dict, results)
get_str('prompt', 'Prompt', loaded_parameter_dict, results)
get_str('negative_prompt', 'Negative Prompt', loaded_parameter_dict, results)
get_list('styles', 'Styles', loaded_parameter_dict, results)
@ -93,11 +93,12 @@ def get_float(key: str, fallback: str | None, source_dict: dict, results: list,
results.append(gr.update())
def get_int(key: str, fallback: str | None, source_dict: dict, results: list, default=None):
def get_image_number(key: str, fallback: str | None, source_dict: dict, results: list, default=None):
try:
h = source_dict.get(key, source_dict.get(fallback, default))
assert h is not None
h = int(h)
h = min(h, modules.config.default_max_image_number)
results.append(h)
except:
results.append(1)