Commit 0cd6399b authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix broken inpainting model

parent 3bd73776
Loading
Loading
Loading
Loading
+0 −3
Original line number Diff line number Diff line
@@ -305,9 +305,6 @@ def load_model(checkpoint_info=None):
        sd_config.model.params.unet_config.params.in_channels = 9
        sd_config.model.params.finetune_keys = None

        # Create a "fake" config with a different name so that we know to unload it when switching models.
        checkpoint_info = checkpoint_info._replace(config=checkpoint_config.replace(".yaml", "-inpainting.yaml"))

    if not hasattr(sd_config.model.params, "use_ema"):
        sd_config.model.params.use_ema = False