lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-28 18:53:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

rework memory management for extras

Browse Source 
now face post-processing uses gpu
close #312
This commit is contained in:
lllyasviel
2024-02-25 20:37:14 -08:00
parent a252bbcf16
commit 6287c73d98
10 changed files with 30 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/hat_model.py
View File

@@ -3,8 +3,9 @@ import sys
from modules import modelloader, devices
from modules.shared import opts
from modules.upscaler import Upscaler, UpscalerData, prepare_free_memory
from modules.upscaler import Upscaler, UpscalerData
from modules.upscaler_utils import upscale_with_model
from modules_forge.forge_util import prepare_free_memory
class UpscalerHAT(Upscaler):