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

Fix a bug where the device wouldn't get set if no cuda-compatible hardware was found

Browse Source
This commit is contained in:
cmdr2 2022-10-27 22:59:55 +05:30
parent 952854f64e
commit 284b95213e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ui/sd_internal/runtime.py
View File

@ -113,6 +113,7 @@ def device_init(device_selection=None):
return return
if not torch.cuda.is_available(): if not torch.cuda.is_available():
print('WARNING: torch.cuda is not available. Using the CPU, but this will be very slow!') print('WARNING: torch.cuda is not available. Using the CPU, but this will be very slow!')
thread_data.device = 'cpu'
return return
device_count = torch.cuda.device_count() device_count = torch.cuda.device_count()
if device_count <= 1 and device_selection == 'auto': if device_count <= 1 and device_selection == 'auto':