Method load_cleaner

diffbir/inference/bfr_loop.py:17–23

(self)

class BFRInferenceLoop(InferenceLoop):

    def load_cleaner(self) -> None:
        self.cleaner: SwinIR = instantiate_from_config(
            OmegaConf.load("configs/inference/swinir.yaml")
        )
        weight = load_model_from_url(MODELS["swinir_face"])
        self.cleaner.load_state_dict(weight, strict=True)
        self.cleaner.eval().to(self.args.device)

    def load_pipeline(self) -> None:
        self.pipeline = SwinIRPipeline(