aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authord8ahazard <d8ahazard@gmail.com>2022-09-30 08:33:06 -0500
committerd8ahazard <d8ahazard@gmail.com>2022-09-30 08:33:06 -0500
commit41f9449d96488017092d229d65f674c2e8252697 (patch)
treedec84f0bc90e258e11b0da8d93eafe98cdd8cd1e
parent5c0c778a65c8f89a85395fb10e32d3b35ea57196 (diff)
Fix LDSR YAML path
-rw-r--r--modules/ldsr_model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/ldsr_model.py b/modules/ldsr_model.py
index 969d1a0d..877e7e73 100644
--- a/modules/ldsr_model.py
+++ b/modules/ldsr_model.py
@@ -24,7 +24,7 @@ class UpscalerLDSR(Upscaler):
def load_model(self, path: str):
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.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)
try: