mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-04-27 09:41:31 +00:00
revise space
This commit is contained in:
@@ -106,6 +106,9 @@ class GPUObject:
|
|||||||
def GPU(gpu_objects=None, manual_load=False):
|
def GPU(gpu_objects=None, manual_load=False):
|
||||||
gpu_objects = gpu_objects or []
|
gpu_objects = gpu_objects or []
|
||||||
|
|
||||||
|
if not isinstance(gpu_objects, (list, tuple)):
|
||||||
|
gpu_objects = [gpu_objects]
|
||||||
|
|
||||||
def decorator(func):
|
def decorator(func):
|
||||||
@functools.wraps(func)
|
@functools.wraps(func)
|
||||||
def wrapper(*args, **kwargs):
|
def wrapper(*args, **kwargs):
|
||||||
|
|||||||
Reference in New Issue
Block a user