fix problems related to checkpoint/VAE switching in XY plot

    def __enter__(self):

        self.CLIP_stop_at_last_layers = opts.CLIP_stop_at_last_layers

        self.hypernetwork = opts.sd_hypernetwork

        self.model = shared.sd_model

        self.vae = opts.sd_vae

    def __exit__(self, exc_type, exc_value, tb):

        modules.sd_models.reload_model_weights(self.model)

        opts.data["sd_vae"] = self.vae

        modules.sd_vae.reload_vae_weights(self.model)

        modules.sd_models.reload_model_weights()

        modules.sd_vae.reload_vae_weights()

        hypernetwork.load_hypernetwork(self.hypernetwork)

        hypernetwork.apply_strength()