diff --git a/src/diffusers/pipelines/flux/pipeline_flux_fill.py b/src/diffusers/pipelines/flux/pipeline_flux_fill.py index ed8623e31733..020d354f3d70 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_fill.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_fill.py @@ -966,3 +966,6 @@ def __call__( return (image,) return FluxPipelineOutput(images=image) + + def parameters(self): + return list(self.transformer.parameters()) + list(self.vae.parameters()) + list(self.text_encoder.parameters()) + list(self.text_encoder_2.parameters())