extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-02 19:49:15 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use the correct device name when moving the model to cpu

Browse Source
This commit is contained in:
cmdr2 2022-11-20 00:43:38 +05:30 committed by GitHub
parent a3463274ee
commit 2cf8b2a453
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ui/sd_internal/runtime.py
View File

@ -245,9 +245,10 @@ def unload_models():
def move_to_cpu(model):
if thread_data.device != "cpu":
mem = torch.cuda.memory_allocated() / 1e6
d = torch.device(thread_data.device)
mem = torch.cuda.memory_allocated(d) / 1e6
model.to("cpu")
while torch.cuda.memory_allocated() / 1e6 >= mem:
while torch.cuda.memory_allocated(d) / 1e6 >= mem:
time.sleep(1)
def load_model_gfpgan():