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

prompt expansion v2

Browse Source
This commit is contained in:
lvmin
2023-09-11 02:48:36 -07:00
parent e6557a4ded
commit 0f658a97f7
6 changed files with 28 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/async_worker.py
+8 -3
View File
@@ -18,6 +18,7 @@ def worker():
from modules.sdxl_styles import apply_style_negative, apply_style_positive, aspect_ratios
from modules.private_logger import log
from modules.expansion import safe_str
from modules.util import join_prompts
try:
async_gradio_app = shared.gradio_root
@@ -81,12 +82,16 @@ def worker():
outputs.append(['preview', (5, f'Preparing positive text #{i + 1} ...', None)])
current_seed = seed + i
p_txt = apply_style_positive(style_selction, prompt)
expansion_weight = 0.35
suffix = pipeline.expansion(p_txt, current_seed)
suffix = pipeline.expansion(prompt, current_seed)
suffix = f'({suffix}:{expansion_weight})'
print(f'[Prompt Expansion] New suffix: {suffix}')
p_txt = safe_str(p_txt) + suffix
p_txt = apply_style_positive(style_selction, prompt)
p_txt = safe_str(p_txt)
p_txt = join_prompts(p_txt, suffix)
tasks.append(dict(
prompt=prompt,