sample solution #10640

[Himasnhu-AT]

#10640

This pull request includes a new method to the FluxPipeline class in the src/diffusers/pipelines/flux/pipeline_flux_fill.py file. The addition of the parameters method allows for load distribution using torch.nn.parallel.dataparallel

• src/diffusers/pipelines/flux/pipeline_flux_fill.py: Added a parameters method to return a list of parameters from transformer, vae, text_encoder, and text_encoder_2.

Important

Wasn't able to test using testcases because im almost out of mem on my PC

USECASE:

pipe = FluxFillPipeline( ... )
pipe = torch.nn.DataParallel(pipe)