lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-26 01:09:10 +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
extensions-builtin/LDSR/scripts/ldsr_model.py
View File

@@ -1,7 +1,8 @@
import os
from modules.modelloader import load_file_from_url
from modules.upscaler import Upscaler, UpscalerData, prepare_free_memory
from modules.upscaler import Upscaler, UpscalerData
from modules_forge.forge_util import prepare_free_memory
from ldsr_model_arch import LDSR
from modules import shared, script_callbacks, errors
import sd_hijack_autoencoder # noqa: F401

3
extensions-builtin/SwinIR/scripts/swinir_model.py
View File

@@ -5,7 +5,8 @@ import torch
from PIL import Image
from modules import devices, modelloader, script_callbacks, shared, upscaler_utils
from modules.upscaler import Upscaler, UpscalerData, prepare_free_memory
from modules.upscaler import Upscaler, UpscalerData
from modules_forge.forge_util import prepare_free_memory
SWINIR_MODEL_URL = "https://github.com/JingyunLiang/SwinIR/releases/download/v0.0/003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth"