Fix size requirements for 1x AdaCode models (#267)

chaiNNer-org · May 20, 2024 · 2612710 · 2612710
1 parent 2a2cda7
commit 2612710
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/spandrel_extra_arches/spandrel_extra_arches/architectures/AdaCode/__init__.py b/libs/spandrel_extra_arches/spandrel_extra_arches/architectures/AdaCode/__init__.py
@@ -133,7 +133,7 @@ def load(self, state_dict: StateDict) -> ImageModelDescriptor[AdaCode]:
             use_residual=use_residual,
         )
 
-        multiple_of = {2: 32, 4: 16}.get(scale_factor, 8)
+        multiple_of = {1: 64, 2: 32, 4: 16}.get(scale_factor, 8)
 
         return ImageModelDescriptor(
             model,