Update __init__.py

chaiNNer-org · Jan 9, 2025 · b336f0d · b336f0d
1 parent 568d0e9
commit b336f0d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/libs/spandrel/spandrel/architectures/ESRGAN/__init__.py b/libs/spandrel/spandrel/architectures/ESRGAN/__init__.py
@@ -216,6 +216,7 @@ def load(self, state_dict: StateDict) -> ImageModelDescriptor[ESRGAN]:
         if shuffle_factor:
             in_nc //= shuffle_factor**2
             scale //= shuffle_factor
+            tags.append("unshuffle")
 
         return ImageModelDescriptor(
             model,