From fc2a6567dadb7f01caa13886b1f9707b68f7ea9a Mon Sep 17 00:00:00 2001
From: Marc-Andre Ferland <madrang@gmail.com>
Date: Fri, 28 Oct 2022 22:51:04 -0400
Subject: [PATCH] Moved import before use of runtime.thread_data.device

---
 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 e984c6a9..01d7bf0a 100644
--- a/ui/sd_internal/task_manager.py
+++ b/ui/sd_internal/task_manager.py
@@ -204,13 +204,13 @@ def preload_model(ckpt_file_path=None, vae_file_path=None):
         print(traceback.format_exc())
 
 def thread_get_next_task():
+    from . import runtime
     if not manager_lock.acquire(blocking=True, timeout=LOCK_TIMEOUT):
         print('Render thread on device', runtime.thread_data.device, 'failed to acquire manager lock.')
         return None
     if len(tasks_queue) <= 0:
         manager_lock.release()
         return None
-    from . import runtime
     task = None
     try:  # Select a render task.
         for queued_task in tasks_queue: