diff --git a/modules/async_worker.py b/modules/async_worker.py index 790ec118..24f5b6f7 100644 --- a/modules/async_worker.py +++ b/modules/async_worker.py @@ -1,6 +1,9 @@ import threading import os -from modules.patch import PatchSettings, patch_settings +from modules.patch import PatchSettings, patch_settings, patch_all + +patch_all() + class AsyncTask: def __init__(self, args): diff --git a/modules/core.py b/modules/core.py index 957a9152..7a29d988 100644 --- a/modules/core.py +++ b/modules/core.py @@ -1,8 +1,3 @@ -from modules.patch import patch_all - -patch_all() - - import os import einops import torch