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

Only return valid data for alive threads.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-30 01:38:32 -04:00
parent eb994716e6
commit c687091ce9
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ui/sd_internal/task_manager.py
View File

@ -356,7 +356,11 @@ def get_devices():
try: try:
device_dict = {} device_dict = {}
for rthread in render_threads: for rthread in render_threads:
if not rthread.is_alive():
continue
weak_data = weak_thread_data.get(rthread) weak_data = weak_thread_data.get(rthread)
if not weak_data or not 'device' in weak_data or not 'device_name' in weak_data:
continue
device_dict.update({weak_data['device']:weak_data['device_name']}) device_dict.update({weak_data['device']:weak_data['device_name']})
return device_dict return device_dict
finally: finally: