Files
ai-toolkit/extensions_built_in/diffusion_models/__init__.py
2025-04-13 13:50:11 -06:00

8 lines
153 B
Python

from .chroma import ChromaModel
from .hidream import HidreamModel
AI_TOOLKIT_MODELS = [
# put a list of models here
ChromaModel, HidreamModel
]