print

Author: sayakpaul

src/diffusers/pipelines/flux/pipeline_flux.py

@@ -776,7 +776,9 @@ def __call__(
             image = self.image_processor.postprocess(image, output_type=output_type)
 
         # Offload all models
+        print("before maybe")
         self.maybe_free_model_hooks()
+        print("after maybe")
 
         if not return_dict:
             return (image,)