Bump version to 0.2.0

2026-04-29 00:31:16 +03:00
parent a5399c0470
commit 6717bce271
1 changed files with 7 additions and 9 deletions
@@ -16,7 +16,12 @@ try:
    from .model import DLAANet
 except ImportError:
    from model import DLAANet
-    
+
+try:
+    from .utils import rgb_luma
+except ImportError:
+    from utils import rgb_luma
+
 try:
    from .presets import (
        PRESETS,
@@ -44,7 +49,7 @@ try:
    import comfy.model_management as mm
 except ImportError:
    mm = None
-    
+
 try:
    from comfy.utils import ProgressBar
 except ImportError:
@@ -53,13 +58,6 @@ except ImportError:
 log = logging.getLogger("VideoTAADLAA")


-LUMA_WEIGHTS = (0.2126, 0.7152, 0.0722)
-
-def rgb_luma(x: torch.Tensor) -> torch.Tensor:
-    r, g, b = LUMA_WEIGHTS
-    return r * x[:, 0:1] + g * x[:, 1:2] + b * x[:, 2:3]
-
-
 class TAAState:
    def __init__(self,
        variance_gamma=1.5,