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

Fixed bug in task_manager.is_alive and added way to check for first device.

Browse Source
This commit is contained in:
Marc-Andre Ferland
2022-10-16 23:06:41 -04:00
parent 994d62ac65
commit 41bfb96b6b
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
ui/sd_internal/runtime.py
View File

@ -149,12 +149,10 @@ def device_init(device_selection=None):
thread_data.device = 'cpu'
def is_first_cuda_device(device):
if thread_data.device == 0 or thread_data.device == '0':
return True
if thread_data.device == 'cuda' or thread_data.device == 'cuda:0':
return True
if thread_data.device == torch.device(0):
return True
if device is None: return False
if device == 0 or device == '0': return True
if device == 'cuda' or device == 'cuda:0': return True
if device == torch.device(0): return True
return False
def load_model_ckpt():