Update pipeline.py
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -866,6 +866,7 @@ class MatForgerPipeline(DiffusionPipeline, FromSingleFileMixin):
|
|
| 866 |
else:
|
| 867 |
image = latents
|
| 868 |
|
|
|
|
| 869 |
image = postprocess(image, output_type=output_type)
|
| 870 |
|
| 871 |
# Offload all models
|
|
|
|
| 866 |
else:
|
| 867 |
image = latents
|
| 868 |
|
| 869 |
+
image.to(torch.float32)
|
| 870 |
image = postprocess(image, output_type=output_type)
|
| 871 |
|
| 872 |
# Offload all models
|