Upload extensions_built_in/diffusion_models/__init__.py with huggingface_hub

extensions_built_in/diffusion_models/__init__.py

@@ -0,0 +1,24 @@
+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
+
+AI_TOOLKIT_MODELS = [
+    # put a list of models here
+    ChromaModel,
+    ChromaRadianceModel,
+    HidreamModel,
+    HidreamE1Model,
+    FLiteModel,
+    OmniGen2Model,
+    FluxKontextModel,
+    Wan225bModel,
+    Wan2214bI2VModel,
+    Wan2214bModel,
+    QwenImageModel,
+    QwenImageEditModel,
+    QwenImageEditPlusModel,
+]