aboutsummaryrefslogtreecommitdiff
path: root/modules/modelloader.py
diff options
context:
space:
mode:
authorvictorca25 <victorca25@users.noreply.github.com>2022-10-30 12:52:50 +0100
committervictorca25 <victor@Victors-MBP.lan>2022-10-30 12:54:06 +0100
commitc9bb33dd43dbb9479ff1b70351df14508c89ac60 (patch)
treef1d516669f810f26dc4206c0dea5bd7041224879 /modules/modelloader.py
parent17a2076f72562b428052ee3fc8c43d19c03ecd1e (diff)
add resrgan 8x, allow use 1x and up to 8x extra models, move BSRGAN model, add nearest
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)