From d5e923da50f5d2a157def74aca00ceece98fac53 Mon Sep 17 00:00:00 2001
From: Joss Reeves <joseph.reeves@gmail.com>
Date: Thu, 10 Apr 2025 19:34:29 -0400
Subject: [PATCH] fixed the upscaler by updating to use the default torch
 device and the modern syntax for torch

---
 nodes/functions_upscale.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes/functions_upscale.py b/nodes/functions_upscale.py
index 2c25f03..5efe6d1 100644
--- a/nodes/functions_upscale.py
+++ b/nodes/functions_upscale.py
@@ -54,7 +54,7 @@ def upscale_with_model(upscale_model, image):
             if tile < 128:
                 raise e
 
-    upscale_model.cpu()
+    upscale_model.to(device)
     s = torch.clamp(s.movedim(-3,-1), min=0, max=1.0)
     return s