Add vae slicing and tiling to flux pipeline

[iamzoltan]

This is an attempt to add enable_vae_slicing and enable_vae_tiling to pipeline_flux. I have tested this locally and it seems to work. Looking to add flux to our roster of models, and this functionality would help.

@yiyixuxu and @asomoza

[yiyixuxu]

thanks!

maybe we should have a DITMixin now? cc @sayakpaul @DN6

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

[sayakpaul]

@yiyixuxu agreed!

@iamzoltan thanks for your contributions here. Did you notice any particular scenarios where enabling slicing and tiling have been helpful (perhaps when operating at higher resolutions on consumer GPU cards)?

[iamzoltan]

@sayakpaul my pleasure! It was helpful on consumer GPU cards. We rely on slicing and tiling for other pipelines from diffusers