diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-11-29 17:06:15 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-11-29 17:06:15 +0300 |
| commit | 4b3c5bc24bffdf429c463a465763b3077fe55eb8 (patch) | |
| tree | f05e78133cd52619d56fc99a12eceaade4b15ac4 | |
| parent | 0b5dcb3d7ce397ad38312dbfc70febe7bb42dcc3 (diff) | |
| parent | ac90cf38c6b55d57d37923aa1fe86c7374e32d0b (diff) | |
Merge remote-tracking branch 'pattontim/safetensors'
| -rw-r--r-- | .gitignore | 1 | ||||
| -rw-r--r-- | modules/modelloader.py | 1 |
2 files changed, 2 insertions, 0 deletions
@@ -1,5 +1,6 @@ __pycache__ *.ckpt +*.safetensors *.pth /ESRGAN/* /SwinIR/* 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) |