From 3bdc90451ac9ce462c445a45ec1352e1a722c2ba Mon Sep 17 00:00:00 2001 From: Marc-Andre Ferland Date: Wed, 19 Oct 2022 04:34:54 -0400 Subject: [PATCH] Dont preload on cpu. --- ui/sd_internal/task_manager.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ui/sd_internal/task_manager.py b/ui/sd_internal/task_manager.py index 9348fe18..da88cfc9 100644 --- a/ui/sd_internal/task_manager.py +++ b/ui/sd_internal/task_manager.py @@ -207,7 +207,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()