Update pipeline.py

pipeline.py

@@ -62,7 +62,8 @@ def postprocess(
         output_type = "np"
 
     image = image.detach().cpu()
-
+    image = image.to(torch.float32)
+    
     if output_type == "latent":
         return image
 
@@ -413,7 +414,7 @@ class MatForgerPipeline(DiffusionPipeline, FromSingleFileMixin):
                 "Provide either `prompt` or `prompt_embeds`. Cannot leave both `prompt` and `prompt_embeds` undefined."
             )
         elif prompt is not None and (
-            not isinstance(prompt, str) and not isinstance(prompt, list)
+            not isinstance(prompt, str) and not isinstance(prompt, list) or not isinstance(prompt, Image.Image)
         ):
             raise ValueError(
                 f"`prompt` has to be of type `str` or `list` but is {type(prompt)}"