Commit e3a973a6 authored by Aarni Koskela's avatar Aarni Koskela
Browse files

Add TODO comments to sus model loads

parent 0afbc0c2
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -119,6 +119,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler):
    def load_model(self, path: str):
        device = devices.get_device_for('scunet')
        if path.startswith("http"):
            # TODO: this doesn't use `path` at all?
            filename = load_file_from_url(self.model_url, model_dir=self.model_download_path, file_name=f"{self.name}.pth")
        else:
            filename = path
+1 −0
Original line number Diff line number Diff line
@@ -151,6 +151,7 @@ class UpscalerESRGAN(Upscaler):

    def load_model(self, path: str):
        if path.startswith("http"):
            # TODO: this doesn't use `path` at all?
            filename = modelloader.load_file_from_url(
                url=self.model_url,
                model_dir=self.model_download_path,