mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-02-08 06:29:56 +00:00
18 lines
408 B
Python
18 lines
408 B
Python
from .chroma import ChromaModel
|
|
from .hidream import HidreamModel, HidreamE1Model
|
|
from .f_light import FLiteModel
|
|
from .omnigen2 import OmniGen2Model
|
|
from .flux_kontext import FluxKontextModel
|
|
from .wan22 import Wan22Model
|
|
|
|
AI_TOOLKIT_MODELS = [
|
|
# put a list of models here
|
|
ChromaModel,
|
|
HidreamModel,
|
|
HidreamE1Model,
|
|
FLiteModel,
|
|
OmniGen2Model,
|
|
FluxKontextModel,
|
|
Wan22Model,
|
|
]
|