Add support for Wan2.2 5B

2025-07-29 05:31:54 -06:00
parent e55116d8c9
commit ca7c5c950b
11 changed files with 1241 additions and 92 deletions
--- a/extensions_built_in/diffusion_models/init.py
+++ b/extensions_built_in/diffusion_models/init.py
@@ -3,13 +3,15 @@ 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
+    ChromaModel,
+    HidreamModel,
+    HidreamE1Model,
+    FLiteModel,
+    OmniGen2Model,
+    FluxKontextModel,
+    Wan22Model,
 ]