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

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