open-mmlab/PowerPaint

About training code

Tangcj1226 opened this issue · 9 comments

Thanks for your great work!
Can you publish the training code ?
I want to train it on SDXL.

+1

hi @WenmuZhou @Tangcj1226 @wangupupup @zhLawliet @lovejing0306 @RunpuWei

Thank you for your interest and being so patient :)
We have released our training configurations and scripts for both train_ppt1_sd15.py for unet-based version and train_ppt2_bn.py for brushnet-based version.

hi @WenmuZhou @Tangcj1226 @wangupupup @zhLawliet @lovejing0306 @RunpuWei

Thank you for your interest and being so patient :) We have released our training configurations and scripts for both train_ppt1_sd15.py for unet-based version and train_ppt2_bn.py for brushnet-based version.

wow, thank you!

very cool, does it support sdxl?will it support flux dev?

Can this script be used for fine-tuning the trained PowerPaint model with LoRA?