This commit is contained in:
lvmin 2023-08-10 21:07:09 -07:00
parent 12a480552f
commit 9992b88c0e
2 changed files with 2 additions and 3 deletions

View File

@ -3,7 +3,6 @@ import os
import torch
from modules.path import modelfile_path, lorafile_path
from modules.cv2win32 import close_all_preview
xl_base_filename = os.path.join(modelfile_path, 'sd_xl_base_1.0.safetensors')
@ -44,6 +43,4 @@ def process(positive_prompt, negative_prompt, steps, switch, width, height, imag
images = core.image_to_numpy(decoded_latent)
close_all_preview()
return images

View File

@ -3,6 +3,7 @@ import random
from modules.sdxl_styles import apply_style, style_keys, aspect_ratios
from modules.default_pipeline import process
from modules.cv2win32 import close_all_preview
def generate_clicked(prompt, negative_prompt, style_selction, performance_selction,
@ -29,6 +30,7 @@ def generate_clicked(prompt, negative_prompt, style_selction, performance_selcti
seed += 1
results += imgs
close_all_preview()
return results