from .chroma import ChromaModel
from .hidream import HidreamModel

AI_TOOLKIT_MODELS = [
    # put a list of models here
    ChromaModel, HidreamModel
]