From 398bc80dec38bfd0ca932fea07e09a1b4916fd35 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sun, 24 Aug 2025 15:48:44 +0800
Subject: [PATCH] Update model_loader_utils.py

---
 model_loader_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/model_loader_utils.py b/model_loader_utils.py
index dc9872b..3365c24 100644
--- a/model_loader_utils.py
+++ b/model_loader_utils.py
@@ -171,6 +171,8 @@ def tensortopil_list_upscale(tensor_in,width,height):
     return img_list
 
 def tensortolist(tensor_in,width,height):
+    if tensor_in is None:
+        return None
     d1, _, _, _ = tensor_in.size()
     if d1 == 1:
         tensor_list = [nomarl_tensor_upscale(tensor_in,width,height)]