Files
ai-toolkit/extensions_built_in/diffusion_models/__init__.py
T

8 lines
153 B
Python

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