This commit is contained in:
lllyasviel
2023-08-15 09:01:25 -07:00
committed by GitHub
parent 6d4e1d22e7
commit 5e5a2d428f
6 changed files with 23 additions and 5 deletions
+4 -1
View File
@@ -14,6 +14,7 @@ def worker():
import random
import modules.default_pipeline as pipeline
import modules.path
import modules.patch
from PIL import Image
from modules.sdxl_styles import apply_style, aspect_ratios
@@ -30,11 +31,13 @@ def worker():
def handler(task):
prompt, negative_prompt, style_selction, performance_selction, \
aspect_ratios_selction, image_number, image_seed, base_model_name, refiner_model_name, \
aspect_ratios_selction, image_number, image_seed, sharpness, base_model_name, refiner_model_name, \
l1, w1, l2, w2, l3, w3, l4, w4, l5, w5 = task
loras = [(l1, w1), (l2, w2), (l3, w3), (l4, w4), (l5, w5)]
modules.patch.sharpness = sharpness
pipeline.refresh_base_model(base_model_name)
pipeline.refresh_refiner_model(refiner_model_name)
pipeline.refresh_loras(loras)