holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Missing .lower() cause CUDA:0 to fail check where cuda:0 works.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-29 04:33:14 -04:00
parent e818f5a93f
commit 03757632cf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/server.py
View File

@ -96,7 +96,7 @@ def setConfig(config):
has_first_cuda_device = False has_first_cuda_device = False
for device in gpu_devices: for device in gpu_devices:
if not task_manager.is_first_cuda_device(device): continue if not task_manager.is_first_cuda_device(device.lower()): continue
has_first_cuda_device = True has_first_cuda_device = True
break break
if len(gpu_devices) > 0 and not has_first_cuda_device: if len(gpu_devices) > 0 and not has_first_cuda_device: