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

Fixed calling get_device_delta with a single cuda device inside config.json at boot.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-11-14 22:14:03 -05:00
parent ffe0eb1544
commit a4a24b1a1a
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ui/sd_internal/device_manager.py
View File

@ -13,10 +13,12 @@ def get_device_delta(render_devices, active_devices):
active_devices: ['cpu', 'cuda:N'...] active_devices: ['cpu', 'cuda:N'...]
''' '''
if render_devices is not None: if render_devices in ('cpu', 'auto'):
if render_devices in ('cpu', 'auto'): render_devices = [render_devices]
elif render_devices is not None:
if isinstance(render_devices, str):
render_devices = [render_devices] render_devices = [render_devices]
elif isinstance(render_devices, list) and len(render_devices) > 0: if isinstance(render_devices, list) and len(render_devices) > 0:
render_devices = list(filter(lambda x: x.startswith('cuda:'), render_devices)) render_devices = list(filter(lambda x: x.startswith('cuda:'), render_devices))
if len(render_devices) == 0: if len(render_devices) == 0:
raise Exception('Invalid render_devices value in config.json. Valid: {"render_devices": ["cuda:0", "cuda:1"...]}, or {"render_devices": "cpu"} or {"render_devices": "auto"}') raise Exception('Invalid render_devices value in config.json. Valid: {"render_devices": ["cuda:0", "cuda:1"...]}, or {"render_devices": "cpu"} or {"render_devices": "auto"}')