mirror of
https://github.com/easydiffusion/easydiffusion.git
synced 2025-04-27 21:08:54 +02:00
Merge pull request #420 from madrang/beta
Missing .lower() cause CUDA:0 to fail check where cuda:0 works.
This commit is contained in:
commit
cf04738594
@ -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:
|
||||||
|
Loading…
Reference in New Issue
Block a user