aboutsummaryrefslogtreecommitdiffstats
path: root/modules/modelloader.py
diff options
context:
space:
mode:
authorTim Patton <38817597+pattontim@users.noreply.github.com>2022-11-19 19:49:22 +0000
committerTim Patton <38817597+pattontim@users.noreply.github.com>2022-11-19 19:49:22 +0000
commitac7ecd2d847bf4e3a9503db0f2a291e32b82302c (patch)
tree1f38e9560f797955024deadd9fe0dade61bf0368 /modules/modelloader.py
parent47a44c7e421b98ca07e92dbf88769b04c9e28f86 (diff)
downloadstable-diffusion-webui-gfx803-ac7ecd2d847bf4e3a9503db0f2a291e32b82302c.tar.gz
stable-diffusion-webui-gfx803-ac7ecd2d847bf4e3a9503db0f2a291e32b82302c.tar.bz2
stable-diffusion-webui-gfx803-ac7ecd2d847bf4e3a9503db0f2a291e32b82302c.zip
Label and load SD .safetensors model files
Diffstat (limited to 'modules/modelloader.py')
-rw-r--r--modules/modelloader.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/modelloader.py b/modules/modelloader.py
index e4a6f8ac..7d2f0ade 100644
--- a/modules/modelloader.py
+++ b/modules/modelloader.py
@@ -82,6 +82,7 @@ def cleanup_models():
src_path = models_path
dest_path = os.path.join(models_path, "Stable-diffusion")
move_files(src_path, dest_path, ".ckpt")
+ move_files(src_path, dest_path, ".safetensors")
src_path = os.path.join(root_path, "ESRGAN")
dest_path = os.path.join(models_path, "ESRGAN")
move_files(src_path, dest_path)