Spaces:

KingNish
/

Realtime-FLUX

Running on Zero

KingNish commited on Sep 17

Commit

8ab26aa

•

1 Parent(s): b7bdba8

Update custom_pipeline.py

Files changed (1) hide show

custom_pipeline.py CHANGED Viewed

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

             torch.cuda.empty_cache()
         # Final image
+        return self._decode_latents_to_image(latents, height, width, output_type)
         self.maybe_free_model_hooks()
         torch.cuda.empty_cache()