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