Commit 74d249f6 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

Merge branch 'release_candidate' into dev

parents 5e4a0e3d 94754c60
Loading
Loading
Loading
Loading
+5 −2
Original line number Diff line number Diff line
@@ -498,6 +498,11 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
    stored_opts = {k: opts.data[k] for k in p.override_settings.keys()}

    try:
        # if no checkpoint override or the override checkpoint can't be found, remove override entry and load opts checkpoint
        if sd_models.checkpoint_alisases.get(p.override_settings.get('sd_model_checkpoint')) is None:
            p.override_settings.pop('sd_model_checkpoint', None)
            sd_models.reload_model_weights()

        for k, v in p.override_settings.items():
            setattr(opts, k, v)

@@ -514,8 +519,6 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
        if p.override_settings_restore_afterwards:
            for k, v in stored_opts.items():
                setattr(opts, k, v)
                if k == 'sd_model_checkpoint':
                    sd_models.reload_model_weights()

                if k == 'sd_vae':
                    sd_vae.reload_vae_weights()
+1 −1
Original line number Diff line number Diff line
@@ -86,7 +86,7 @@ def apply_checkpoint(p, x, xs):
    info = modules.sd_models.get_closet_checkpoint_match(x)
    if info is None:
        raise RuntimeError(f"Unknown checkpoint: {x}")
    modules.sd_models.reload_model_weights(shared.sd_model, info)
    p.override_settings['sd_model_checkpoint'] = info.hash


def confirm_checkpoints(p, xs):