mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-02-24 08:43:57 +00:00
b2353a4911
underlying gallery object changed with gradio update, old code broken, new code more simple added check for attempt to upscale grid removed redundant check already covered by second assert
132 lines
5.3 KiB
Python
132 lines
5.3 KiB
Python
import json
|
|
from contextlib import closing
|
|
|
|
import modules.scripts
|
|
from modules import processing, infotext_utils
|
|
from modules.infotext_utils import create_override_settings_dict, parse_generation_parameters
|
|
from modules.shared import opts
|
|
import modules.shared as shared
|
|
from modules.ui import plaintext_to_html
|
|
from PIL import Image
|
|
import gradio as gr
|
|
from modules_forge import main_thread
|
|
|
|
|
|
def txt2img_create_processing(id_task: str, request: gr.Request, prompt: str, negative_prompt: str, prompt_styles, n_iter: int, batch_size: int, cfg_scale: float, distilled_cfg_scale: float, height: int, width: int, enable_hr: bool, denoising_strength: float, hr_scale: float, hr_upscaler: str, hr_second_pass_steps: int, hr_resize_x: int, hr_resize_y: int, hr_checkpoint_name: str, hr_sampler_name: str, hr_scheduler: str, hr_prompt: str, hr_negative_prompt, override_settings_texts, *args, force_enable_hr=False):
|
|
override_settings = create_override_settings_dict(override_settings_texts)
|
|
|
|
if force_enable_hr:
|
|
enable_hr = True
|
|
|
|
p = processing.StableDiffusionProcessingTxt2Img(
|
|
outpath_samples=opts.outdir_samples or opts.outdir_txt2img_samples,
|
|
outpath_grids=opts.outdir_grids or opts.outdir_txt2img_grids,
|
|
prompt=prompt,
|
|
styles=prompt_styles,
|
|
negative_prompt=negative_prompt,
|
|
batch_size=batch_size,
|
|
n_iter=n_iter,
|
|
cfg_scale=cfg_scale,
|
|
distilled_cfg_scale=distilled_cfg_scale,
|
|
width=width,
|
|
height=height,
|
|
enable_hr=enable_hr,
|
|
denoising_strength=denoising_strength,
|
|
hr_scale=hr_scale,
|
|
hr_upscaler=hr_upscaler,
|
|
hr_second_pass_steps=hr_second_pass_steps,
|
|
hr_resize_x=hr_resize_x,
|
|
hr_resize_y=hr_resize_y,
|
|
hr_checkpoint_name=None if hr_checkpoint_name == 'Use same checkpoint' else hr_checkpoint_name,
|
|
hr_sampler_name=None if hr_sampler_name == 'Use same sampler' else hr_sampler_name,
|
|
hr_scheduler=None if hr_scheduler == 'Use same scheduler' else hr_scheduler,
|
|
hr_prompt=hr_prompt,
|
|
hr_negative_prompt=hr_negative_prompt,
|
|
override_settings=override_settings,
|
|
)
|
|
|
|
p.scripts = modules.scripts.scripts_txt2img
|
|
p.script_args = args
|
|
|
|
p.user = request.username
|
|
|
|
if shared.opts.enable_console_prompts:
|
|
print(f"\ntxt2img: {prompt}", file=shared.progress_print_out)
|
|
|
|
return p
|
|
|
|
|
|
def txt2img_upscale_function(id_task: str, request: gr.Request, gallery, gallery_index, generation_info, *args):
|
|
assert len(gallery) > 0, 'No image to upscale'
|
|
assert 0 <= gallery_index < len(gallery), f'Bad image index: {gallery_index}'
|
|
|
|
# catch situation where user tries to hires-fix the grid: probably a mistake, results can be bad aspect ratio - just don't do it
|
|
if opts.return_grid == True and 0 == gallery_index:
|
|
return gallery, generation_info, 'Unable to upscale the grid image.', ''
|
|
|
|
p = txt2img_create_processing(id_task, request, *args, force_enable_hr=True)
|
|
p.batch_size = 1
|
|
p.n_iter = 1
|
|
# txt2img_upscale attribute that signifies this is called by txt2img_upscale
|
|
p.txt2img_upscale = True
|
|
|
|
geninfo = json.loads(generation_info)
|
|
|
|
image_info = gallery[gallery_index]
|
|
p.firstpass_image = infotext_utils.image_from_url_text(image_info)
|
|
|
|
parameters = parse_generation_parameters(geninfo.get('infotexts')[gallery_index], [])
|
|
p.seed = parameters.get('Seed', -1)
|
|
p.subseed = parameters.get('Variation seed', -1)
|
|
|
|
p.override_settings['save_images_before_highres_fix'] = False
|
|
|
|
with closing(p):
|
|
processed = modules.scripts.scripts_txt2img.run(p, *p.script_args)
|
|
|
|
if processed is None:
|
|
processed = processing.process_images(p)
|
|
|
|
shared.total_tqdm.clear()
|
|
|
|
new_gallery = []
|
|
for i, image in enumerate(gallery):
|
|
if i == gallery_index:
|
|
geninfo["infotexts"][gallery_index: gallery_index+1] = processed.infotexts
|
|
new_gallery.extend(processed.images)
|
|
else:
|
|
new_gallery.append(image)
|
|
|
|
geninfo["infotexts"][gallery_index] = processed.info
|
|
|
|
return new_gallery, json.dumps(geninfo), plaintext_to_html(processed.info), plaintext_to_html(processed.comments, classname="comments")
|
|
|
|
|
|
def txt2img_function(id_task: str, request: gr.Request, *args):
|
|
p = txt2img_create_processing(id_task, request, *args)
|
|
|
|
with closing(p):
|
|
processed = modules.scripts.scripts_txt2img.run(p, *p.script_args)
|
|
|
|
if processed is None:
|
|
processed = processing.process_images(p)
|
|
|
|
shared.total_tqdm.clear()
|
|
|
|
generation_info_js = processed.js()
|
|
if opts.samples_log_stdout:
|
|
print(generation_info_js)
|
|
|
|
if opts.do_not_show_images:
|
|
processed.images = []
|
|
|
|
return processed.images + processed.extra_images, generation_info_js, plaintext_to_html(processed.info), plaintext_to_html(processed.comments, classname="comments")
|
|
|
|
|
|
def txt2img_upscale(id_task: str, request: gr.Request, gallery, gallery_index, generation_info, *args):
|
|
return main_thread.run_and_wait_result(txt2img_upscale_function, id_task, request, gallery, gallery_index, generation_info, *args)
|
|
|
|
|
|
def txt2img(id_task: str, request: gr.Request, *args):
|
|
return main_thread.run_and_wait_result(txt2img_function, id_task, request, *args)
|