From 6720c99f7082dc855059ad4afd6b3cb45b62bc1f Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 1 Oct 2022 16:53:19 +0200 Subject: Fix seed, better progress bar, fix euler_a for batch size > 1 --- pipelines/stable_diffusion/clip_guided_stable_diffusion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'pipelines/stable_diffusion') diff --git a/pipelines/stable_diffusion/clip_guided_stable_diffusion.py b/pipelines/stable_diffusion/clip_guided_stable_diffusion.py index ddf7ce1..eff74b5 100644 --- a/pipelines/stable_diffusion/clip_guided_stable_diffusion.py +++ b/pipelines/stable_diffusion/clip_guided_stable_diffusion.py @@ -254,10 +254,10 @@ class CLIPGuidedStableDiffusion(DiffusionPipeline): noise_pred = None if isinstance(self.scheduler, EulerAScheduler): sigma = t.reshape(1) - sigma_in = torch.cat([sigma] * 2) + sigma_in = torch.cat([sigma] * latent_model_input.shape[0]) # noise_pred = model(latent_model_input,sigma_in,uncond_embeddings, text_embeddings,guidance_scale) noise_pred = CFGDenoiserForward(self.unet, latent_model_input, sigma_in, - text_embeddings, guidance_scale, DSsigmas=self.scheduler.DSsigmas) + text_embeddings, guidance_scale, quantize=True, DSsigmas=self.scheduler.DSsigmas) # noise_pred = self.unet(latent_model_input, sigma_in, encoder_hidden_states=text_embeddings).sample else: # predict the noise residual -- cgit v1.2.3-54-g00ecf