From 09141ee1a8f51df7ef0a4b0c2ffbd053a3628b74 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sun, 28 May 2023 20:00:03 +0300 Subject: [PATCH] Fix int64 with UniPC && Add OneAPI version logging --- installer.py | 3 ++- modules/sd_samplers_compvis.py | 5 ++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/installer.py b/installer.py index 25a6e18c2..8a36342d3 100644 --- a/installer.py +++ b/installer.py @@ -289,7 +289,8 @@ def check_torch(): log.info(f'Torch {torch.__version__}') if args.use_ipex and allow_ipex: import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import - log.info(f'Torch backend: Intel OneAPI {torch.__version__}') + log.info(f'Torch backend: Intel IPEX {ipex.__version__}') + log.info(f'{os.popen("icpx --version").read().rstrip()}') log.info(f'Torch detected GPU: {torch.xpu.get_device_name("xpu")} VRAM {round(torch.xpu.get_device_properties("xpu").total_memory / 1024 / 1024)}') elif torch.cuda.is_available() and (allow_cuda or allow_rocm): if torch.version.cuda and allow_cuda: diff --git a/modules/sd_samplers_compvis.py b/modules/sd_samplers_compvis.py index 98b0a3614..0fb411111 100644 --- a/modules/sd_samplers_compvis.py +++ b/modules/sd_samplers_compvis.py @@ -97,7 +97,10 @@ class VanillaStableDiffusionSampler: unconditional_conditioning = unconditional_conditioning[:, :cond.shape[1]] if self.mask is not None: - img_orig = self.sampler.model.q_sample(self.init_latent, ts) + if shared.cmd_opts.use_ipex: + img_orig = self.sampler.model.q_sample(self.init_latent, ts.type(torch.int64)) + else: + img_orig = self.sampler.model.q_sample(self.init_latent, ts) x = img_orig * self.mask + self.nmask * x # Wrap the image conditioning back up since the DDIM code can accept the dict directly.