models/__init__.py

from .model import NextDiT_2B_patch2
from .model import NextDiT_2B_GQA_patch2