Commit eec540b2 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

repair non-latent upscaling broken for SDXL

parent 77761e7b
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -1119,6 +1119,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
            decoded_samples = torch.from_numpy(np.array(batch_images))
            decoded_samples = decoded_samples.to(shared.device)
            decoded_samples = 2. * decoded_samples - 1.
            decoded_samples = decoded_samples.to(shared.device, dtype=devices.dtype_vae)

            samples = self.sd_model.get_first_stage_encoding(self.sd_model.encode_first_stage(decoded_samples))