fix rcan detection keys

chaiNNer-org · Jan 9, 2025 · 8ba2502 · 8ba2502
1 parent e88b0a5
commit 8ba2502
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/libs/spandrel/spandrel/architectures/RCAN/__init__.py b/libs/spandrel/spandrel/architectures/RCAN/__init__.py
@@ -12,15 +12,15 @@ def __init__(self):
             id="RCAN",
             detect=KeyCondition.has_any(
                 KeyCondition.has_all(
-                    # "head.0.weight",
-                    # "tail.1.weight",
-                    # "body.0.body.0.body.0.weight",
+                    "head.0.weight",
+                    "tail.1.weight",
+                    "body.0.body.0.body.0.weight",
                     "body.0.body.0.body.3.conv_du.0.weight",
                 ),
                 KeyCondition.has_all(
-                    # "head.1.weight",
-                    # "tail.1.weight",
-                    # "body.0.body.0.body.0.weight",
+                    "head.1.weight",
+                    "tail.1.weight",
+                    "body.0.body.0.body.0.weight",
                     "body.0.body.0.body.3.conv_du.0.weight",
                 ),
             ),