Add initial support for FLUX.2

extensions_built_in/diffusion_models/__init__.py

@@ -5,6 +5,7 @@ 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
 
 AI_TOOLKIT_MODELS = [
     # put a list of models here
@@ -21,4 +22,5 @@ AI_TOOLKIT_MODELS = [
     QwenImageModel,
     QwenImageEditModel,
     QwenImageEditPlusModel,
+    Flux2Model,
 ]