mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
10 lines
248 B
Python
10 lines
248 B
Python
from .chroma import ChromaModel
|
|
from .hidream import HidreamModel
|
|
from .f_light import FLiteModel
|
|
from .omnigen2 import OmniGen2Model
|
|
|
|
AI_TOOLKIT_MODELS = [
|
|
# put a list of models here
|
|
ChromaModel, HidreamModel, FLiteModel, OmniGen2Model
|
|
]
|