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()