From 389e3397ecc8b2e47217d0428a8291d0c5253e9b Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Thu, 27 Oct 2022 23:17:41 +0530 Subject: [PATCH] Preload the model even in the CPU mode --- ui/sd_internal/task_manager.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ui/sd_internal/task_manager.py b/ui/sd_internal/task_manager.py index efc53c08..5a565c64 100644 --- a/ui/sd_internal/task_manager.py +++ b/ui/sd_internal/task_manager.py @@ -251,8 +251,7 @@ def thread_render(device): weak_thread_data[threading.current_thread()] = { 'device': runtime.thread_data.device } - if runtime.thread_data.device != 'cpu': - preload_model() + preload_model() current_state = ServerStates.Online while True: task_cache.clean()