From c7e822ca3aa995acbee1fa9447235519b4cd61a5 Mon Sep 17 00:00:00 2001
From: tntwise <tntwise3@gmail.com>
Date: Thu, 16 May 2024 04:19:37 +0000
Subject: [PATCH] whoops pt 2

---
 src/torch/UpscaleImageTensorRT.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/torch/UpscaleImageTensorRT.py b/src/torch/UpscaleImageTensorRT.py
index b1b7d29a..5a622b39 100644
--- 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():