comfy pls

__init__.py

@@ -32,10 +32,16 @@ original_validate = execution.validate_prompt
 
 
 @wraps(original_validate)
-async def hijack_validate(prompt_id: int, prompt: dict) -> bool:
+async def hijack_validate(*args):
+    for arg in args:
+        if isinstance(arg, dict):
+            prompt = arg
+            break
+
     if not find_node(prompt):
         disable_recenter()
-    return await original_validate(prompt_id, prompt)
+
+    return await original_validate(*args)
 
 
 execution.validate_prompt = hijack_validate

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-diffusion-cg"
 description = "Color Grading for Stable Diffusion"
-version = "1.2.0"
+version = "1.3.0"
 license = { text = "MIT License" }
 dependencies = []