from .model import NextDiT_2B_patch2