Update pipeline_stable_diffusion_3_ipa.py
Browse files
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
|