flake/packages/codeformer/root_dir.patch
matthewcroughan 5615f114d7 init
2023-02-11 22:54:48 +00:00

14 lines
411 B
Diff

diff --git a/facelib/utils/misc.py b/facelib/utils/misc.py
index 0240187..0570161 100644
--- a/facelib/utils/misc.py
+++ b/facelib/utils/misc.py
@@ -10,7 +10,7 @@ from urllib.parse import urlparse
import gdown
-ROOT_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+ROOT_DIR = f'{os.getcwd()}/models/facelib'
def download_pretrained_models(file_ids, save_path_root):