formatting

chaiNNer-org · Jul 13, 2024 · 741e975 · 741e975
1 parent 698386e
commit 741e975
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/libs/spandrel/spandrel/architectures/__arch_helpers/dysample.py b/libs/spandrel/spandrel/architectures/__arch_helpers/dysample.py
@@ -65,7 +65,10 @@ def forward(self, x):
             .to(x.device, non_blocking=True)
         )
         normalizer = torch.tensor(
-            [W, H], dtype=x.dtype, device=x.device, pin_memory=False # pin_memory was originally True
+            [W, H],
+            dtype=x.dtype,
+            device=x.device,
+            pin_memory=False,  # pin_memory was originally True
         ).view(1, 2, 1, 1, 1)
         coords = 2 * (coords + offset) / normalizer - 1