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

Update upscaler.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 13:23:04 -08:00
parent d44608f5f5
commit cf82cdb018
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/upscaler.py
View File

@@ -11,6 +11,7 @@ from ldm_patched.modules import model_management
def prepare_free_memory():
model_management.free_memory(memory_required=1024*1024*3, device=model_management.get_torch_device())
print('Upscale script is freed memory successfully.')
LANCZOS = (Image.Resampling.LANCZOS if hasattr(Image, 'Resampling') else Image.LANCZOS)