lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-25 09:13:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Label and load SD .safetensors model files

Browse Source
This commit is contained in:
Tim Patton
2022-11-19 14:49:22 -05:00
parent 47a44c7e42
commit ac7ecd2d84
4 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/modelloader.py
View File

@@ -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)