Merge remote-tracking branch 'pattontim/safetensors'

3 years ago · 4b3c5bc24b
parent 0b5dcb3d7c ac90cf38c6
commit 4b3c5bc24b
2 changed files with 2 additions and 0 deletions
--- a/.gitignore
+++ b/.gitignore
@ -1,5 +1,6 @@
 __pycache__
 *.ckpt
+*.safetensors
 *.pth
 /ESRGAN/*
 /SwinIR/*
--- 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)