Commit 41f9449d authored by d8ahazard's avatar d8ahazard

Fix LDSR YAML path

parent 5c0c778a
...@@ -24,7 +24,7 @@ class UpscalerLDSR(Upscaler): ...@@ -24,7 +24,7 @@ class UpscalerLDSR(Upscaler):
def load_model(self, path: str): def load_model(self, path: str):
model = load_file_from_url(url=self.model_url, model_dir=self.model_path, model = load_file_from_url(url=self.model_url, model_dir=self.model_path,
file_name="model.pth", progress=True) file_name="model.pth", progress=True)
yaml = load_file_from_url(url=self.model_url, model_dir=self.model_path, yaml = load_file_from_url(url=self.yaml_url, model_dir=self.model_path,
file_name="project.yaml", progress=True) file_name="project.yaml", progress=True)
try: try:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment