michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Remove LDSR project.yaml if too large.

Browse source 
If we accidentally downloaded the wrong file and saved it as project.yaml, this will delete it so it can be re-downloaded.
This commit is contained in:
d8ahazard 2022-09-30 08:41:25 -05:00
parent 41f9449d96
commit 64c6b13312
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
modules/ldsr_model.py
View file

@ -22,6 +22,13 @@ class UpscalerLDSR(Upscaler):
self.scalers = [scaler_data]
def load_model(self, path: str):
# Remove incorrect project.yaml file if too big
yaml_path = os.path.join(self.model_path, "project.yaml")
if os.path.exists(yaml_path):
statinfo = os.stat(yaml_path)
if statinfo.st_size <= 10485760:
print("Removing invalid LDSR YAML file.")
os.remove(yaml_path)
model = load_file_from_url(url=self.model_url, model_dir=self.model_path,
file_name="model.pth", progress=True)
yaml = load_file_from_url(url=self.yaml_url, model_dir=self.model_path,