revise space

2026-01-30 12:59:47 +00:00 · 2024-08-19 00:05:09 -07:00
parent 3bef4e331a
commit 00115ae02a
1 changed files with 3 additions and 0 deletions
--- 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):