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: