extensions_built_in/diffusion_models/__init__.py

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

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