extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-10 11:08:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #512 from cmdr2/beta

Browse Source 
Use the correct device name when moving the model to cpu
This commit is contained in:
cmdr2 2022-11-20 00:50:58 +05:30 committed by GitHub
commit 8bcdb205ed
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): def move_to_cpu(model):
if thread_data.device != "cpu": 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") model.to("cpu")
while torch.cuda.memory_allocated() / 1e6 >= mem: while torch.cuda.memory_allocated(d) / 1e6 >= mem:
time.sleep(1) time.sleep(1)
def load_model_gfpgan(): def load_model_gfpgan():