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

2023-07-30 16:16:31 +03:00 · 2023-07-30 16:16:31 +03:00 · 02038036ff
parent cc53db6652
commit 02038036ff
1 changed files with 1 additions and 1 deletions
--- a/modules/processing.py
+++ b/modules/processing.py
@ -1069,7 +1069,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
            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