File size: 43 Bytes
4af8a26
1
2
from .dit import DiT, DiT_XL_2, DiT_XL_2x2