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

Refactor qwen5b model code to be qwen 5b specific

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-06 10:54:56 -06:00
parent 5d8922fca2
commit 14ccf2f3ce
4 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
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 Wan22Model
from .wan22 import Wan225bModel
from .qwen_image import QwenImageModel
AI_TOOLKIT_MODELS = [
@@ -14,6 +14,6 @@ AI_TOOLKIT_MODELS = [
FLiteModel,
OmniGen2Model,
FluxKontextModel,
Wan22Model,
Wan225bModel,
QwenImageModel,
]