from .ace_step import AceStep15Model, AceStep15XLModel
from .yue2 import YuE2AudioModel

AI_TOOLKIT_MODELS = [
    # put a list of models here
    AceStep15Model,
    AceStep15XLModel,
    YuE2AudioModel,
]