From 22a11769fa02d88ac692f5ca0f728ef944b9f2e7 Mon Sep 17 00:00:00 2001 From: Marc-Andre Ferland Date: Thu, 27 Oct 2022 21:57:50 -0400 Subject: [PATCH] Enable preload on cpu when no other devices are alive. --- ui/sd_internal/task_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ui/sd_internal/task_manager.py b/ui/sd_internal/task_manager.py index 1d0535ab..034e0f69 100644 --- a/ui/sd_internal/task_manager.py +++ b/ui/sd_internal/task_manager.py @@ -250,7 +250,7 @@ def thread_render(device): weak_thread_data[threading.current_thread()] = { 'device': runtime.thread_data.device } - if runtime.thread_data.device != 'cpu': + if runtime.thread_data.device != 'cpu' or is_alive() == 1: preload_model() current_state = ServerStates.Online while True: