Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Made device agnostic

Browse Source
This commit is contained in:
wywywywy 2022-12-10 14:07:27 +00:00
parent 6df316c881
commit 1581d5a167
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
extensions-builtin/LDSR/ldsr_model_arch.py
View File

@ -110,6 +110,7 @@ class LDSR:
down_sample_method = 'Lanczos' down_sample_method = 'Lanczos'
gc.collect() gc.collect()
if torch.cuda.is_available:
torch.cuda.empty_cache() torch.cuda.empty_cache()
im_og = image im_og = image
@ -147,7 +148,9 @@ class LDSR:
del model del model
gc.collect() gc.collect()
if torch.cuda.is_available:
torch.cuda.empty_cache() torch.cuda.empty_cache()
return a return a
@ -162,7 +165,7 @@ def get_cond(selected_path):
c = rearrange(c, '1 c h w -> 1 h w c') c = rearrange(c, '1 c h w -> 1 h w c')
c = 2. * c - 1. c = 2. * c - 1.
c = c.to(torch.device("cuda")) c = c.to(shared.device)
example["LR_image"] = c example["LR_image"] = c
example["image"] = c_up example["image"] = c_up