the train script for dit model

[TtuHamg]

Thank you very much for your open-source work. Does the train_diffit.py file you provided support fine-tuning the dit model? Could you provide details and parameters for fine-tuning the dit model?

[tanghengjian]

I think this training work, just change the cross attention function by adding some gamma_fnn Parameter instead of using the stable diffusion unet cross attention. So i wonder what's relation with DiT model.