Label and load SD .safetensors model files

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

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)