extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-26 04:41:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Dont preload on cpu.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-19 04:34:54 -04:00
parent a036b2981a
commit 3bdc90451a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ui/sd_internal/task_manager.py
View File

@ -207,7 +207,8 @@ 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
} }
preload_model() if runtime.thread_data.device != 'cpu':
preload_model()
current_state = ServerStates.Online current_state = ServerStates.Online
while True: while True:
task_cache.clean() task_cache.clean()