extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-18 11:20:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

Don't try to start cuda:0 if auto used cpu mode.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-29 00:36:26 -04:00
parent 4675da4d16
commit 902ccbd203
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

@ -407,7 +407,7 @@ display_warning = False
if task_manager.is_alive() <= 0: # Either no defauls or no devices after loading config. if task_manager.is_alive() <= 0: # Either no defauls or no devices after loading config.
# Select best GPU device using free memory, if more than one device. # Select best GPU device using free memory, if more than one device.
if task_manager.start_render_thread('auto'): # Detect best device for renders if task_manager.start_render_thread('auto'): # Detect best device for renders
if task_manager.is_alive(0) <= 0: # is not cuda:0 if task_manager.is_alive(0) <= 0 and task_manager.is_alive('cpu') <= 0: # is not cuda:0 and auto didnt use CPU.
if task_manager.start_render_thread('cuda'): # An other cuda device is better and cuda:0 is missing, start it... if task_manager.start_render_thread('cuda'): # An other cuda device is better and cuda:0 is missing, start it...
display_warning = True # And warn user to update settings... display_warning = True # And warn user to update settings...
else: else: