from .chroma import ChromaModel
from .hidream import HidreamModel
from .f_light import FLiteModel
from .omnigen2 import OmniGen2Model

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