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

Only default to cpu on auto or current.

Browse Source 
Not when a specific device was requested.
This commit is contained in:
Marc-Andre Ferland 2022-10-28 01:08:00 -04:00
parent 0a569146a8
commit 71c6beadb4
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ui/sd_internal/runtime.py
View File

@ -112,9 +112,12 @@ def device_init(device_selection=None):
thread_data.device = 'cpu'
return
if not torch.cuda.is_available():
if device_selection == 'auto' or device_selection == 'current':
print('WARNING: torch.cuda is not available. Using the CPU, but this will be very slow!')
thread_data.device = 'cpu'
return
else:
raise EnvironmentError('torch.cuda is not available.')
device_count = torch.cuda.device_count()
if device_count <= 1 and device_selection == 'auto':
device_selection = 'current' # Use 'auto' only when there is more than one compatible device found.