diff options
| author | unknown <mcgpapu@gmail.com> | 2022-12-25 08:03:55 +0000 |
|---|---|---|
| committer | unknown <mcgpapu@gmail.com> | 2022-12-25 08:03:55 +0000 |
| commit | 876da1259965130603f2a7fea505cfa0fce09e2e (patch) | |
| tree | ccb8b89d64480a4bd224b311702ffeb13b8fe754 /extensions-builtin/LDSR/scripts/ldsr_model.py | |
| parent | d6fdfde9d70f1b86b696240fb0a0c8f2a4d024f6 (diff) | |
| parent | c6f347b81f584b6c0d44af7a209983284dbb52d2 (diff) | |
| download | stable-diffusion-webui-gfx803-876da1259965130603f2a7fea505cfa0fce09e2e.tar.gz stable-diffusion-webui-gfx803-876da1259965130603f2a7fea505cfa0fce09e2e.tar.bz2 stable-diffusion-webui-gfx803-876da1259965130603f2a7fea505cfa0fce09e2e.zip | |
Merge branch 'master' of github.com:AUTOMATIC1111/stable-diffusion-webui
Diffstat (limited to 'extensions-builtin/LDSR/scripts/ldsr_model.py')
| -rw-r--r-- | extensions-builtin/LDSR/scripts/ldsr_model.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/extensions-builtin/LDSR/scripts/ldsr_model.py b/extensions-builtin/LDSR/scripts/ldsr_model.py index 29d5f94e..b8cff29b 100644 --- a/extensions-builtin/LDSR/scripts/ldsr_model.py +++ b/extensions-builtin/LDSR/scripts/ldsr_model.py @@ -25,6 +25,7 @@ class UpscalerLDSR(Upscaler): yaml_path = os.path.join(self.model_path, "project.yaml") old_model_path = os.path.join(self.model_path, "model.pth") new_model_path = os.path.join(self.model_path, "model.ckpt") + safetensors_model_path = os.path.join(self.model_path, "model.safetensors") if os.path.exists(yaml_path): statinfo = os.stat(yaml_path) if statinfo.st_size >= 10485760: @@ -33,8 +34,11 @@ class UpscalerLDSR(Upscaler): if os.path.exists(old_model_path): print("Renaming model from model.pth to model.ckpt") os.rename(old_model_path, new_model_path) - model = load_file_from_url(url=self.model_url, model_dir=self.model_path, - file_name="model.ckpt", progress=True) + if os.path.exists(safetensors_model_path): + model = safetensors_model_path + else: + model = load_file_from_url(url=self.model_url, model_dir=self.model_path, + file_name="model.ckpt", progress=True) yaml = load_file_from_url(url=self.yaml_url, model_dir=self.model_path, file_name="project.yaml", progress=True) |