mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-27 00:49:47 +00:00
5b5aadadb8
* WIP, adding support for LTX2 * Training on images working * Fix loading comfy models * Handle converting and deconverting lora so it matches original format * Reworked ui to habdle ltx and propert dataset default overwriting. * Update the way lokr saves to it is more compatable with comfy * Audio loading and synchronization/resampling is working * Add audio to training. Does it work? Maybe, still testing. * Fixed fps default issue for sound * Have ui set fps for accurate audio mapping on ltx * Added audio procession options to the ui for ltx * Clean up requirements
31 lines
826 B
Python
31 lines
826 B
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
|
|
from .z_image import ZImageModel
|
|
from .ltx2 import LTX2Model
|
|
|
|
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,
|
|
]
|