Commit 570f42af authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

possible fix for FP16 VAE failing in img2img SDXL

parent 0198eaec

modules/processing.py

+1 −1
Original line number Diff line number Diff line
@@ -1303,7 +1303,7 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing): 


        image = torch.from_numpy(batch_images)

        image = 2. * image - 1.

        image = image.to(shared.device)

        image = image.to(shared.device, dtype=devices.dtype_vae)



        self.init_latent = self.sd_model.get_first_stage_encoding(self.sd_model.encode_first_stage(image))

CRA Git | Maintained and supported by SUSTech CRA and CCSE