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

Fixed is_alive to work with devices that can change name after init.

Browse Source
This commit is contained in:
Marc-Andre Ferland
2022-10-18 20:33:37 -04:00
parent cfd6751777
commit fcdb086daf
3 changed files with 36 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ui/sd_internal/runtime.py
View File

@ -130,7 +130,7 @@ def device_init(device_selection=None):
device_selection = device_selection.lower()
if device_selection.startswith('gpu:'):
device_selection = int(device_selection[4:])
if device_selection != 'current' and device_selection != 'gpu':
if device_selection != 'cuda' and device_selection != 'current' and device_selection != 'gpu':
if device_select(device_selection):
if isinstance(device_selection, int):
print(f'Setting GPU:{device_selection} as active')