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

Put back the check to only preload on GPU

Browse Source
This commit is contained in:
cmdr2 2022-10-28 00:04:33 +05:30
parent 5d8bda1178
commit 0dfaf9159d
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ui/sd_internal/task_manager.py
View File

@ -251,6 +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: