File size: 80 Bytes
5aadc08
 
1
2
3
# from .model import DiT_Llama_5B_patch2
from .model import DiT_Llama_2B_patch2