Merge remote-tracking branch 'pattontim/safetensors'

This commit is contained in:
AUTOMATIC 2022-11-29 17:06:15 +03:00
commit 4b3c5bc24b
2 changed files with 2 additions and 0 deletions

1
.gitignore vendored
View File

@ -1,5 +1,6 @@
__pycache__
*.ckpt
*.safetensors
*.pth
/ESRGAN/*
/SwinIR/*

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)