aboutsummaryrefslogtreecommitdiffstats
path: root/modules/modelloader.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-11-02 11:09:38 +0000
committerGitHub <noreply@github.com>2022-11-02 11:09:38 +0000
commite359268be9936db1c16c78adf544d622d33d1bfb (patch)
tree15d41dc2832ea849192eb46f22a26cb2255f76f9 /modules/modelloader.py
parentbb21a4cb35986d95ec63ace48ce13b75a776f5a5 (diff)
parentc9bb33dd43dbb9479ff1b70351df14508c89ac60 (diff)
downloadstable-diffusion-webui-gfx803-e359268be9936db1c16c78adf544d622d33d1bfb.tar.gz
stable-diffusion-webui-gfx803-e359268be9936db1c16c78adf544d622d33d1bfb.tar.bz2
stable-diffusion-webui-gfx803-e359268be9936db1c16c78adf544d622d33d1bfb.zip
Merge pull request #3976 from victorca25/esrgan_fea
multiple trivial changes for "extras" models
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)