From 7864a464f540842822d3a9c4d6ab1cd4e34698d4 Mon Sep 17 00:00:00 2001
From: MMP0 <28616020+MMP0@users.noreply.github.com>
Date: Wed, 21 Feb 2024 08:05:33 +0900
Subject: [PATCH] Fix task queue not showing up in 1.8.0

---
 agent_scheduler/task_helpers.py | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/agent_scheduler/task_helpers.py b/agent_scheduler/task_helpers.py
index 1829272..27efdd4 100644
--- a/agent_scheduler/task_helpers.py
+++ b/agent_scheduler/task_helpers.py
@@ -12,11 +12,9 @@ from PIL import Image, ImageOps, ImageChops, ImageEnhance, ImageFilter, PngImage
 from numpy import ndarray
 from torch import Tensor
 
-from modules import sd_samplers, scripts, shared, sd_vae, images
+from modules import sd_samplers, scripts, shared, sd_vae, images, txt2img, img2img
 from modules.generation_parameters_copypaste import create_override_settings_dict
 from modules.sd_models import CheckpointInfo, get_closet_checkpoint_match
-from modules.txt2img import txt2img
-from modules.img2img import img2img
 from modules.api.models import (
     StableDiffusionTxt2ImgProcessingAPI,
     StableDiffusionImg2ImgProcessingAPI,
@@ -241,11 +239,11 @@ def map_controlnet_args_to_api_task_args(args: Dict):
 
 
 def map_ui_task_args_list_to_named_args(args: List, is_img2img: bool):
-    args_name = []
     if is_img2img:
-        args_name = inspect.getfullargspec(img2img).args
+        fn = getattr(img2img, "img2img_create_processing", None) or img2img.img2img
     else:
-        args_name = inspect.getfullargspec(txt2img).args
+        fn = getattr(txt2img, "txt2img_create_processing", None) or txt2img.txt2img
+    args_name = inspect.getfullargspec(fn).args
 
     # SD WebUI 1.5.0 has new request arg
     if "request" in args_name:
@@ -277,11 +275,11 @@ def map_ui_task_args_list_to_named_args(args: List, is_img2img: bool):
 
 
 def map_named_args_to_ui_task_args_list(named_args: Dict, script_args: List, is_img2img: bool):
-    args_name = []
     if is_img2img:
-        args_name = inspect.getfullargspec(img2img).args
+        fn = getattr(img2img, "img2img_create_processing", None) or img2img.img2img
     else:
-        args_name = inspect.getfullargspec(txt2img).args
+        fn = getattr(txt2img, "txt2img_create_processing", None) or txt2img.txt2img
+    args_name = inspect.getfullargspec(fn).args
 
     sampler_name = named_args.get("sampler_name", None)
     if sampler_name is not None: