extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-14 06:28:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Moved import before use of runtime.thread_data.device

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-28 22:51:04 -04:00
parent ef6f491d94
commit fc2a6567da
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/sd_internal/task_manager.py
View File

@ -204,13 +204,13 @@ def preload_model(ckpt_file_path=None, vae_file_path=None):
print(traceback.format_exc()) print(traceback.format_exc())
def thread_get_next_task(): def thread_get_next_task():
from . import runtime
if not manager_lock.acquire(blocking=True, timeout=LOCK_TIMEOUT): if not manager_lock.acquire(blocking=True, timeout=LOCK_TIMEOUT):
print('Render thread on device', runtime.thread_data.device, 'failed to acquire manager lock.') print('Render thread on device', runtime.thread_data.device, 'failed to acquire manager lock.')
return None return None
if len(tasks_queue) <= 0: if len(tasks_queue) <= 0:
manager_lock.release() manager_lock.release()
return None return None
from . import runtime
task = None task = None
try: # Select a render task. try: # Select a render task.
for queued_task in tasks_queue: for queued_task in tasks_queue: