提交 64c6b133 编写于 作者: D d8ahazard

Remove LDSR project.yaml if too large.

If we accidentally downloaded the wrong file and saved it as project.yaml, this will delete it so it can be re-downloaded.
上级 41f9449d
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册