extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-23 23:49:11 +01: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

3
ui/sd_internal/task_manager.py
View File

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