Commit 435fd211 authored by d8ahazard's avatar d8ahazard
Browse files

Fixes, cleanup.

parent d7374179
Loading
Loading
Loading
Loading
+4 −2
Original line number Diff line number Diff line
@@ -101,8 +101,10 @@ class LDSR:
        print("Foo")
        down_sample_rate = target_scale / 4
        print(f"Downsample rate is {down_sample_rate}")
        width_downsampled_pre = width_og * down_sample_rate
        height_downsampled_pre = height_og * down_sample_method
        wd = width_og * down_sample_rate
        hd = height_og * down_sample_rate
        width_downsampled_pre = int(wd)
        height_downsampled_pre = int(hd)

        if down_sample_rate != 1:
            print(
+0 −14
Original line number Diff line number Diff line
@@ -118,11 +118,9 @@ def load_upscalers():
    for cls in Upscaler.__subclasses__():
        name = cls.__name__
        module_name = cls.__module__
        print(f"Class: {name} and {module_name}")
        module = importlib.import_module(module_name)
        class_ = getattr(module, name)
        cmd_name = f"{name.lower().replace('upscaler', '')}-models-path"
        print(f"CMD Name: {cmd_name}")
        opt_string = None
        try:
            opt_string = shared.opts.__getattr__(cmd_name)
@@ -130,18 +128,6 @@ def load_upscalers():
            pass
        scaler = class_(opt_string)
        for child in scaler.scalers:
            print(f"Appending {child.name}")
            datas.append(child)

    shared.sd_upscalers = datas

    # for scaler in subclasses:
    #     print(f"Found scaler: {type(scaler).__name__}")
    #     try:
    #         scaler = scaler()
    #         for child in scaler.scalers:
    #             print(f"Appending {child.name}")
    #             datas.append[child]
    #     except:
    #         pass
    # shared.sd_upscalers = datas
+1 −1
Original line number Diff line number Diff line
@@ -52,7 +52,7 @@ class UpscalerSwinIR(Upscaler):

    def load_model(self, path, scale=4):
        if "http" in path:
            dl_name = "%s%s" % (self.name.replace(" ", "_"), ".pth")
            dl_name = "%s%s" % (self.model_name.replace(" ", "_"), ".pth")
            filename = load_file_from_url(url=path, model_dir=self.model_path, file_name=dl_name, progress=True)
        else:
            filename = path