diff options
| author | Max Audron <audron@cocaine.farm> | 2023-01-25 17:15:42 +0100 |
|---|---|---|
| committer | Max Audron <audron@cocaine.farm> | 2023-01-27 14:44:30 +0100 |
| commit | 5eee2ac39863f9e44591b50d0710dd2615416a13 (patch) | |
| tree | 72d90317db4e724a9d9ace4f9e8d8f2d4467d5e9 /modules/gfpgan_model.py | |
| parent | 9beb794e0b0dc1a0f9e89d8e38bd789a8c608397 (diff) | |
add data-dir flag and set all user data directories based on it
Diffstat (limited to 'modules/gfpgan_model.py')
| -rw-r--r-- | modules/gfpgan_model.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/modules/gfpgan_model.py b/modules/gfpgan_model.py index 1e2dbc32..fbe6215a 100644 --- a/modules/gfpgan_model.py +++ b/modules/gfpgan_model.py @@ -6,12 +6,11 @@ import facexlib import gfpgan
import modules.face_restoration
-from modules import shared, devices, modelloader
-from modules.paths import models_path
+from modules import paths, shared, devices, modelloader
model_dir = "GFPGAN"
user_path = None
-model_path = os.path.join(models_path, model_dir)
+model_path = os.path.join(paths.models_path, model_dir)
model_url = "https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth"
have_gfpgan = False
loaded_gfpgan_model = None
|