whoops pt 2

TNTwise · May 16, 2024 · c7e822c · c7e822c
1 parent 1c267cc
commit c7e822c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torch/UpscaleImageTensorRT.py b/src/torch/UpscaleImageTensorRT.py
@@ -75,7 +75,7 @@ def pytorchExportToONNX(self): # Loads model via spandrel, and exports to onnx
         input = torch.rand(1, 3, 256, 256)
         if self.half:
             model.half()
-            input.half()
+            input = input.half()
         with torch.inference_mode():