make it so that VAE NaNs autodetection also works during first pass of hires fix

            return samples

        if self.latent_scale_mode is None:

            decoded_samples = decode_first_stage(self.sd_model, samples)

            decoded_samples = torch.stack(decode_latent_batch(self.sd_model, samples, target_device=devices.cpu, check_for_nans=True))

        else:

            decoded_samples = None