From b60f49ede6ea8e3503d0098f264a1d53d8dc867a Mon Sep 17 00:00:00 2001
From: Ryan Mueller <ryan.dalrymple.mueller@gmail.com>
Date: Thu, 6 Nov 2025 19:02:31 -0500
Subject: [PATCH] Fixed out channel attention

---
 src/Restorer/Cond_NAF.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/Restorer/Cond_NAF.py b/src/Restorer/Cond_NAF.py
index 5dc2d0f..1deaa46 100644
--- a/src/Restorer/Cond_NAF.py
+++ b/src/Restorer/Cond_NAF.py
@@ -281,7 +281,7 @@ def forward(self, input):
 
         x = self.dropout2(x)
 
-        x = self.sca_out(x) * x
+        x = self.sca_out(x, cond) * x
 
 
         return (y + x * self.gamma, cond)