bug fix

LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2024 Haoming
+Copyright (c) 2025 Haoming
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

__init__.py

@@ -1,7 +1,9 @@
+from functools import wraps
+
+import execution
+
 from .normalization import Normalization
 from .recenter import Recenter, RecenterXL, disable_recenter
-from functools import wraps
-import execution
 
 NODE_CLASS_MAPPINGS = {
     "Normalization": Normalization,
@@ -30,12 +32,10 @@ original_validate = execution.validate_prompt
 
 
 @wraps(original_validate)
-def hijack_validate(prompt: dict):
-
+async def hijack_validate(prompt_id: int, prompt: dict) -> bool:
     if not find_node(prompt):
         disable_recenter()
-
-    return original_validate(prompt)
+    return await original_validate(prompt_id, prompt)
 
 
 execution.validate_prompt = hijack_validate

pyproject.toml

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

recenter.py

@@ -1,8 +1,9 @@
 from functools import wraps
-import datetime
-import comfy
+
 import torch
 
+import comfy
+
 RECENTER: float = 0.0
 LUTS: list[float] = None