From f4faf8e40fc0d989aaec7d5732720d9ca11f5d5d Mon Sep 17 00:00:00 2001 From: MSXYZ-GenAI Date: Mon, 20 Apr 2026 14:00:32 +0300 Subject: [PATCH] Fix DLAANet weight loading --- VideoTAADLAA.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/VideoTAADLAA.py b/VideoTAADLAA.py index 577f24b..48dc9c5 100644 --- a/VideoTAADLAA.py +++ b/VideoTAADLAA.py @@ -1,6 +1,6 @@ # Created by MSXYZ (AI-assisted) # Temporal Anti-Aliasing (TAA) + Lightweight DLAA-style Sharpening -# v0.1.1 - Optimized DLAANet, Improved Stability +# v0.1.1 - Fix DLAANet weight loading, Improved Stability import torch import torch.nn as nn @@ -198,7 +198,7 @@ class VideoTAADLAA: # apply residual mostly to luminance to avoid color shifts luma_res = 0.2126 * residual[:, 0:1] + 0.7152 * residual[:, 1:2] + 0.0722 * residual[:, 2:3] - rgb = rgb + (luma_res * dlaa_strength * 4.0) + rgb = rgb + (luma_res * dlaa_strength * 7.0) # slight gamma & contrast adjustment mean_luma = torch.mean(luma_orig, dim=(1,2,3), keepdim=True)