lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 03:01:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix progressbar missing

Browse Source
This commit is contained in:
layerdiffusion
2024-08-18 02:02:15 -07:00
parent 2292f9a100
commit 608af2e64c
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
spaces.py
View File

@@ -5,6 +5,7 @@ initialize_forge()
import os import os
import torch import torch
import inspect import inspect
import functools
from backend import memory_management from backend import memory_management
from diffusers.models import modeling_utils as diffusers_modeling_utils 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 [] gpu_objects = gpu_objects or []
def decorator(func): def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
print("Entering Forge Space GPU ...") print("Entering Forge Space GPU ...")
memory_management.unload_all_models() memory_management.unload_all_models()