Files
ai-toolkit/extensions_built_in/diffusion_models/__init__.py
2025-05-01 11:15:18 -06:00

9 lines
197 B
Python

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
]