fix vae decode in sd demo

microsoft · Nov 4, 2023 · eab1706 · eab1706
1 parent eff8636
commit eab1706
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/python/tools/transformers/models/stable_diffusion/pipeline_txt2img.py b/onnxruntime/python/tools/transformers/models/stable_diffusion/pipeline_txt2img.py
@@ -79,7 +79,7 @@ def _infer(
             latents = self.denoise_latent(latents, text_embeddings, guidance=guidance)
 
             # VAE decode latent
-            images = self.decode_latent(latents)
+            images = self.decode_latent(latents / self.vae_scaling_factor)
 
             torch.cuda.synchronize()
             e2e_toc = time.perf_counter()