From 1aab44cb496b1eb996308dcae56d439d3228d9df Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 22 Jun 2024 19:39:17 -0400 Subject: [PATCH] fix --- modules/processing_args.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/processing_args.py b/modules/processing_args.py index 7f47b6352..91f10c80d 100644 --- a/modules/processing_args.py +++ b/modules/processing_args.py @@ -27,7 +27,8 @@ def task_specific_kwargs(p, model): 'height': 8 * math.ceil(p.height / 8), } elif (sd_models.get_diffusers_task(model) == sd_models.DiffusersTaskType.IMAGE_2_IMAGE or is_img2img_model) and len(getattr(p, 'init_images', [])) > 0: - model.register_to_config(requires_aesthetics_score = False) + if shared.sd_model_type == 'sdxl': + model.register_to_config(requires_aesthetics_score = False) p.ops.append('img2img') task_args = { 'image': p.init_images, @@ -42,7 +43,8 @@ def task_specific_kwargs(p, model): 'strength': p.denoising_strength, } elif (sd_models.get_diffusers_task(model) == sd_models.DiffusersTaskType.INPAINTING or is_img2img_model) and len(getattr(p, 'init_images', [])) > 0: - model.register_to_config(requires_aesthetics_score = False) + if shared.sd_model_type == 'sdxl': + model.register_to_config(requires_aesthetics_score = False) p.ops.append('inpaint') width, height = processing_helpers.resize_init_images(p) task_args = {