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

Enable preload on cpu when no other devices are alive.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-27 21:57:50 -04:00
parent 7dc7ba9977
commit 22a11769fa
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/sd_internal/task_manager.py
View File

@ -250,7 +250,7 @@ def thread_render(device):
weak_thread_data[threading.current_thread()] = { weak_thread_data[threading.current_thread()] = {
'device': runtime.thread_data.device 'device': runtime.thread_data.device
} }
if runtime.thread_data.device != 'cpu': if runtime.thread_data.device != 'cpu' or is_alive() == 1:
preload_model() preload_model()
current_state = ServerStates.Online current_state = ServerStates.Online
while True: while True: