refactor: use central flag for LoRA count

This commit is contained in:
Manuel Schmid
2024-01-29 14:26:56 +01:00
parent 13d0341a02
commit c3ab9f1f30
6 changed files with 11 additions and 8 deletions
+2 -2
View File
@@ -44,7 +44,7 @@ def worker():
from modules.util import remove_empty_str, HWC3, resize_image, \
get_image_shape_ceil, set_image_shape_ceil, get_shape_ceil, resample_image, erode_or_dilate, calculate_sha256
from modules.upscaler import perform_upscale
from modules.flags import Performance, MetadataScheme
from modules.flags import Performance, MetadataScheme, lora_count
try:
async_gradio_app = shared.gradio_root
@@ -134,7 +134,7 @@ def worker():
base_model_name = args.pop()
refiner_model_name = args.pop()
refiner_switch = args.pop()
loras = [[str(args.pop()), float(args.pop())] for _ in range(5)]
loras = [[str(args.pop()), float(args.pop())] for _ in range(lora_count)]
input_image_checkbox = args.pop()
current_tab = args.pop()
uov_method = args.pop()