Upload combined_stable_diffusion.py with huggingface_hub

combined_stable_diffusion.py

@@ -440,7 +440,7 @@ class CombinedStableDiffusionXL(
 
             latents = latents / self.vae.config.scaling_factor
 
-            image = self.vae.decode(latents, return_dict=False)[0]
+            image = self.vae.decode(latents).sample
 
             # cast back to fp16 if needed
             if needs_upcasting:
@@ -449,7 +449,11 @@ class CombinedStableDiffusionXL(
             image = latents
 
         if not output_type == "latent":
-            image = self.image_processor.postprocess(image, output_type=output_type)
+            image = self.image_processor.postprocess(
+                image,
+                output_type=output_type,
+                do_denormalize=[True] * image.shape[0]
+            )
 
         # Offload all models
         self.maybe_free_model_hooks()