From 00115ae02a60e46ea720d4e5c686d74fef8758ca Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Mon, 19 Aug 2024 00:05:09 -0700
Subject: [PATCH] revise space

---
 spaces.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/spaces.py b/spaces.py
index ea0fd7c7..cfcbe393 100644
--- a/spaces.py
+++ b/spaces.py
@@ -106,6 +106,9 @@ class GPUObject:
 def GPU(gpu_objects=None, manual_load=False):
     gpu_objects = gpu_objects or []
 
+    if not isinstance(gpu_objects, (list, tuple)):
+        gpu_objects = [gpu_objects]
+
     def decorator(func):
         @functools.wraps(func)
         def wrapper(*args, **kwargs):