extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-02 11:40:05 +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
if not torch.cuda.is_available():
print('WARNING: torch.cuda is not available. Using the CPU, but this will be very slow!')
thread_data.device = 'cpu'
return
device_count = torch.cuda.device_count()
if device_count <= 1 and device_selection == 'auto':