From 9a151b7defb6bacb5ce67209e728b14922ad7454 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sun, 26 May 2024 13:44:47 -0400
Subject: [PATCH] Fix issue and unpin spandrel package.

---
 comfy_extras/nodes_upscale_model.py | 2 +-
 requirements.txt                    | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/comfy_extras/nodes_upscale_model.py b/comfy_extras/nodes_upscale_model.py
index f0bbba76..03f29446 100644
--- a/comfy_extras/nodes_upscale_model.py
+++ b/comfy_extras/nodes_upscale_model.py
@@ -42,7 +42,7 @@ class ImageUpscaleWithModel:
     def upscale(self, upscale_model, image):
         device = model_management.get_torch_device()
 
-        memory_required = model_management.module_size(upscale_model)
+        memory_required = model_management.module_size(upscale_model.model)
         memory_required += (512 * 512 * 3) * image.element_size() * max(upscale_model.scale, 1.0) * 384.0 #The 384.0 is an estimate of how much some of these models take, TODO: make it more accurate
         memory_required += image.nelement() * image.element_size()
         model_management.free_memory(memory_required, device)
diff --git a/requirements.txt b/requirements.txt
index a1a29218..906b96ed 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,5 +10,7 @@ Pillow
 scipy
 tqdm
 psutil
+
+#non essential dependencies:
 kornia>=0.7.1
-spandrel==0.3.1
\ No newline at end of file
+spandrel