From 608af2e64c3e96454adfd6e93d1e019f234f4e50 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Sun, 18 Aug 2024 02:02:15 -0700
Subject: [PATCH] fix progressbar missing

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

diff --git a/spaces.py b/spaces.py
index 0b1d2264..d335f537 100644
--- a/spaces.py
+++ b/spaces.py
@@ -5,6 +5,7 @@ initialize_forge()
 import os
 import torch
 import inspect
+import functools
 
 from backend import memory_management
 from diffusers.models import modeling_utils as diffusers_modeling_utils
@@ -89,6 +90,7 @@ def GPU(gpu_objects=None, manual_load=False):
     gpu_objects = gpu_objects or []
 
     def decorator(func):
+        @functools.wraps(func)
         def wrapper(*args, **kwargs):
             print("Entering Forge Space GPU ...")
             memory_management.unload_all_models()