1 files changed, 2 insertions, 2 deletions

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