1inkusFace commited on
Commit
8f76135
·
verified ·
1 Parent(s): 5ae2c4a

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +2 -1
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -1297,7 +1297,8 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1297
  else:
1298
  latents = (latents / self.vae.config.scaling_factor) + self.vae.config.shift_factor
1299
 
1300
- image = self.vae.decode(latents, return_dict=False)[0]
 
1301
  image = self.image_processor.postprocess(image, output_type=output_type)
1302
 
1303
  # Offload all models
 
1297
  else:
1298
  latents = (latents / self.vae.config.scaling_factor) + self.vae.config.shift_factor
1299
 
1300
+ image = self.vae.decode(latents.to(torch.float32), return_dict=False)[0]
1301
+
1302
  image = self.image_processor.postprocess(image, output_type=output_type)
1303
 
1304
  # Offload all models