mindspore-lab / mindone

one for all, Optimal generator with No Exception
https://mindspore-lab.github.io/mindone/
Apache License 2.0
354 stars 67 forks source link

DiT:Inference and Train #377

Closed wtomin closed 6 months ago

wtomin commented 7 months ago

What does this PR do?

Adds # (feature)

  1. DiT model model definition in mindone/models/dit.py
  2. Train and inference scripts for DiT model, using class labels as conditions.

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag members/contributors who may be interested in your PR.

@xxx

zhtmike commented 6 months ago

better to separated into two folder/PR if possible

wtomin commented 6 months ago

better to separated into two folder/PR if possible

Fixed.