From d4c9d752f439310de7bebc825334fc81f805fc44 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 13 Jul 2023 23:13:44 +0300 Subject: [PATCH] Fix latent antialias with IPEX --- modules/processing.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/modules/processing.py b/modules/processing.py index b9ceacab0..181e966c8 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -1003,7 +1003,10 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): if latent_scale_mode is not None: for i in range(samples.shape[0]): save_intermediate(samples, i) - samples = torch.nn.functional.interpolate(samples, size=(target_height // opt_f, target_width // opt_f), mode=latent_scale_mode["mode"], antialias=latent_scale_mode["antialias"]) + if devices.backend == 'ipex' and latent_scale_mode["antialias"]: + samples = torch.nn.functional.interpolate(samples.to("cpu"), size=(target_height // opt_f, target_width // opt_f), mode=latent_scale_mode["mode"], antialias=latent_scale_mode["antialias"]).to(devices.get_cuda_device_string()) + else: + samples = torch.nn.functional.interpolate(samples, size=(target_height // opt_f, target_width // opt_f), mode=latent_scale_mode["mode"], antialias=latent_scale_mode["antialias"]) if getattr(self, "inpainting_mask_weight", shared.opts.inpainting_mask_weight) < 1.0: image_conditioning = self.img2img_image_conditioning(decode_first_stage(self.sd_model, samples.to(dtype=devices.dtype_vae)), samples) else: