aboutsummaryrefslogtreecommitdiff
path: root/modules/modelloader.py
diff options
context:
space:
mode:
authorMuhammad Rizqi Nur <rizqinur2010@gmail.com>2022-11-02 20:48:58 +0700
committerMuhammad Rizqi Nur <rizqinur2010@gmail.com>2022-11-02 20:48:58 +0700
commit237e79c77deec1924b3547f49402b9c1e51c9535 (patch)
tree589f9aea7ea27791df575371ff322d99ad3f3dbc /modules/modelloader.py
parentd5ea878b2aa117588d85287cbd8983aa52177df5 (diff)
parent172c4bc09f0866e7dd114068ebe0f9abfe79ef33 (diff)
Merge branch 'master' into gradient-clipping
Diffstat (limited to 'modules/modelloader.py')
-rw-r--r--modules/modelloader.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/modelloader.py b/modules/modelloader.py
index b0f2f33d..e4a6f8ac 100644
--- a/modules/modelloader.py
+++ b/modules/modelloader.py
@@ -85,6 +85,9 @@ def cleanup_models():
src_path = os.path.join(root_path, "ESRGAN")
dest_path = os.path.join(models_path, "ESRGAN")
move_files(src_path, dest_path)
+ src_path = os.path.join(models_path, "BSRGAN")
+ dest_path = os.path.join(models_path, "ESRGAN")
+ move_files(src_path, dest_path, ".pth")
src_path = os.path.join(root_path, "gfpgan")
dest_path = os.path.join(models_path, "GFPGAN")
move_files(src_path, dest_path)