Update custom_pipeline.py

custom_pipeline.py

@@ -155,7 +155,7 @@ class FLUXPipelineWithIntermediateOutputs(FluxPipeline):
             torch.cuda.empty_cache()
 
         # Final image
-        yield self._decode_latents_to_image(latents, height, width, output_type)
+        return self._decode_latents_to_image(latents, height, width, output_type)
         self.maybe_free_model_hooks()
         torch.cuda.empty_cache()