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

Preload the model even in the CPU mode

Browse Source
This commit is contained in:
cmdr2 2022-10-27 23:17:41 +05:30
parent 284b95213e
commit 389e3397ec
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ui/sd_internal/task_manager.py
View File

@ -251,8 +251,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': preload_model()
preload_model()
current_state = ServerStates.Online current_state = ServerStates.Online
while True: while True:
task_cache.clean() task_cache.clean()