gvecchio commited on
Commit
1fffeef
1 Parent(s): 693939a

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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