Update pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -62,7 +62,8 @@ def postprocess(
|
|
62 |
output_type = "np"
|
63 |
|
64 |
image = image.detach().cpu()
|
65 |
-
|
|
|
66 |
if output_type == "latent":
|
67 |
return image
|
68 |
|
@@ -866,7 +867,6 @@ class MatForgerPipeline(DiffusionPipeline, FromSingleFileMixin):
|
|
866 |
else:
|
867 |
image = latents
|
868 |
|
869 |
-
image.to(torch.float32)
|
870 |
image = postprocess(image, output_type=output_type)
|
871 |
|
872 |
# Offload all models
|
|
|
62 |
output_type = "np"
|
63 |
|
64 |
image = image.detach().cpu()
|
65 |
+
image = image.to(torch.float32)
|
66 |
+
|
67 |
if output_type == "latent":
|
68 |
return image
|
69 |
|
|
|
867 |
else:
|
868 |
image = latents
|
869 |
|
|
|
870 |
image = postprocess(image, output_type=output_type)
|
871 |
|
872 |
# Offload all models
|