From 6717bce271ea6e7c535d848dda6b356bfa2b8697 Mon Sep 17 00:00:00 2001 From: MSXYZ-GenAI Date: Wed, 29 Apr 2026 00:31:16 +0300 Subject: [PATCH] Bump version to 0.2.0 --- VideoTAADLAA.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/VideoTAADLAA.py b/VideoTAADLAA.py index a572cc0..1334bbf 100644 --- a/VideoTAADLAA.py +++ b/VideoTAADLAA.py @@ -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,