3e6bd874c4
* Add Anima training support * Update Anima modular training * Use sample guidance for Anima * Fix Anima sampling * Limit Anima LoRA targets * Convert Anima LoRA exports * Fix Anima local loading * Update Anima default model * Pin upstream Anima diffusers * Adjust template defaults to be consistent with other models. Update README --------- Co-authored-by: Jaret Burkett (Ostris) <jaretburkett@gmail.com>
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
from .chroma import ChromaModel, ChromaRadianceModel
|
|
from .hidream import HidreamModel, HidreamE1Model
|
|
from .f_light import FLiteModel
|
|
from .omnigen2 import OmniGen2Model
|
|
from .flux_kontext import FluxKontextModel
|
|
from .wan22 import Wan225bModel, Wan2214bModel, Wan2214bI2VModel
|
|
from .qwen_image import QwenImageModel, QwenImageEditModel, QwenImageEditPlusModel
|
|
from .flux2 import Flux2Model, Flux2Klein4BModel, Flux2Klein9BModel
|
|
from .z_image import ZImageModel
|
|
from .ltx2 import LTX2Model, LTX23Model
|
|
from .zeta_chroma import ZetaChromaModel
|
|
from .ernie_image import ErnieImageModel
|
|
from .nucleus_image import NucleusImageModel
|
|
from .hidream.hidream_o1_model import HidreamO1Model
|
|
from .z_image.z_image_l2p_model import ZImageL2PModel
|
|
from .anima import AnimaModel
|
|
from .ideogram4 import Ideogram4Model
|
|
from .prx_pixel_t2i import PRXPixelT2IModel
|
|
from .krea2 import Krea2Model
|
|
from .boogu_image import BooguImageModel, BooguImageEditModel
|
|
|
|
AI_TOOLKIT_MODELS = [
|
|
# put a list of models here
|
|
ChromaModel,
|
|
ChromaRadianceModel,
|
|
HidreamModel,
|
|
HidreamE1Model,
|
|
FLiteModel,
|
|
OmniGen2Model,
|
|
FluxKontextModel,
|
|
Wan225bModel,
|
|
Wan2214bI2VModel,
|
|
Wan2214bModel,
|
|
QwenImageModel,
|
|
QwenImageEditModel,
|
|
QwenImageEditPlusModel,
|
|
Flux2Model,
|
|
ZImageModel,
|
|
LTX2Model,
|
|
LTX23Model,
|
|
Flux2Klein4BModel,
|
|
Flux2Klein9BModel,
|
|
ZetaChromaModel,
|
|
ErnieImageModel,
|
|
NucleusImageModel,
|
|
HidreamO1Model,
|
|
ZImageL2PModel,
|
|
AnimaModel,
|
|
Ideogram4Model,
|
|
PRXPixelT2IModel,
|
|
Krea2Model,
|
|
BooguImageModel,
|
|
BooguImageEditModel,
|
|
]
|