Add missing property

chaiNNer-org · Jul 13, 2024 · e954fa2 · e954fa2
1 parent ed3230a
commit e954fa2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/libs/spandrel/spandrel/architectures/PLKSR/__arch/RealPLKSR.py b/libs/spandrel/spandrel/architectures/PLKSR/__arch/RealPLKSR.py
@@ -123,6 +123,7 @@ def __init__(
         out_ch: int = 3
 
         self.dysample = dysample
+        self.upscaling_factor = upscaling_factor
 
         if not self.training:
             dropout = 0
@@ -157,6 +158,6 @@ def __init__(
 
     def forward(self, x: torch.Tensor) -> torch.Tensor:
         x = self.feats(x) + self.repeat_op(x)
-        if not self.dysample or (self.dysample and self.upscale != 1):
+        if not self.dysample or (self.dysample and self.upscaling_factor != 1):
             x = self.to_img(x)
         return x