power88/webui-fooocus-prompt-expansion
1
0
Fork 0
mirror of https://github.com/power88/webui-fooocus-prompt-expansion.git synced 2026-04-28 02:01:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

use InputAccordion

Browse Source
This commit is contained in:
w-e-w
2024-06-09 17:51:05 +09:00
parent 483fdc4330
commit c59d9d1032
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/expansion.py
View File

@@ -19,7 +19,7 @@ from huggingface_hub import hf_hub_download
from transformers.generation.logits_process import LogitsProcessorList
from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
from modules import scripts, paths_internal, errors, shared, script_callbacks
from modules.ui_components import FormRow, FormColumn, FormGroup, ToolButton
from modules.ui_components import InputAccordion
def text_encoder_device():
@@ -244,12 +244,8 @@ class FooocusPromptExpansion(scripts.Script):
return scripts.AlwaysVisible
def ui(self, is_img2img):
with gr.Group():
with gr.Accordion("Fooocus Expansion", open=True):
is_enabled = gr.Checkbox(
value=True, label="Enable Expansion", info="Enable Or Disable Expansion ")
seed = gr.Number(
value=0, maximum=63, label="Seed", info="Seed for random number generator")
with InputAccordion(False, label="Fooocus Expansion") as is_enabled:
seed = gr.Number(value=0, maximum=63, label="Seed", info="Seed for random number generator")
return [is_enabled, seed]
def process(self, p, is_enabled, seed):