controlnet training script for flux

[universewill]

controlnet training script for flux

[geroldmeisinger]

https://github.com/XLabs-AI/x-flux/blob/main/train_flux_deepspeed_controlnet.py

[kadirnar]

https://github.com/XLabs-AI/x-flux/blob/main/train_flux_deepspeed_controlnet.py

I tested it this week but multi gpu is not working in parallel. I kept getting cuda memory error.

[kadirnar]

@sayakpaul Could you add this feature to the Diffusers library?

[sayakpaul]

Sorry we don’t have the bandwidth for this on the team. But happy to help if a contributor wants pick it up.

[kadirnar]

Sorry we don’t have the bandwidth for this on the team. But happy to help if a contributor wants pick it up.

I don't know much about train code but I will try. Thanks for your feedback.

[christopher-beckham]

Has it not already been done here? #9175

[kadirnar]

Has it not already been done here? #9175

This is just the inference code. I want the train code.

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.

[sayakpaul]

#9324

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.