diff --git a/ui/easydiffusion/task_manager.py b/ui/easydiffusion/task_manager.py index 3192f183..a3aa607d 100644 --- a/ui/easydiffusion/task_manager.py +++ b/ui/easydiffusion/task_manager.py @@ -24,9 +24,6 @@ from easydiffusion.utils import log from torchruntime.utils import get_device_count, get_device, get_device_name, get_installed_torch_platform from sdkit.utils import is_cpu_device, mem_get_info -from torchruntime.utils import get_device_count, get_device, get_device_name, get_installed_torch_platform -from sdkit.utils import is_cpu_device, mem_get_info - THREAD_NAME_PREFIX = "" ERR_LOCK_FAILED = " failed to acquire lock within timeout." LOCK_TIMEOUT = 15 # Maximum locking time in seconds before failing a task.