Add support for Mage-Flow and Mage-Flow Edit

extensions_built_in/diffusion_models/__init__.py

@@ -18,6 +18,7 @@ from .ideogram4 import Ideogram4Model
 from .prx_pixel_t2i import PRXPixelT2IModel
 from .krea2 import Krea2Model
 from .boogu_image import BooguImageModel, BooguImageEditModel
+from .mageflow import MageFlowModel, MageFlowEditModel
 
 AI_TOOLKIT_MODELS = [
     # put a list of models here
@@ -51,4 +52,6 @@ AI_TOOLKIT_MODELS = [
     Krea2Model,
     BooguImageModel,
     BooguImageEditModel,
+    MageFlowModel,
+    MageFlowEditModel,
 ]