ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-28 10:11:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Wan22 14b training is working, still need tons of testing and some bug fixes

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-14 13:03:27 -06:00
parent be71cc75ce
commit 3413fa537f
8 changed files with 554 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
extensions_built_in/diffusion_models/__init__.py
View File

@@ -3,7 +3,7 @@ from .hidream import HidreamModel, HidreamE1Model
from .f_light import FLiteModel
from .omnigen2 import OmniGen2Model
from .flux_kontext import FluxKontextModel
from .wan22 import Wan225bModel
from .wan22 import Wan225bModel, Wan2214bModel
from .qwen_image import QwenImageModel
AI_TOOLKIT_MODELS = [
@@ -15,5 +15,6 @@ AI_TOOLKIT_MODELS = [
OmniGen2Model,
FluxKontextModel,
Wan225bModel,
Wan2214bModel,
QwenImageModel,
]