[feat] Add ostris / Flex.2-preview pipeline

src/diffusers/__init__.py

@@ -360,6 +360,7 @@
             "EasyAnimateControlPipeline",
             "EasyAnimateInpaintPipeline",
             "EasyAnimatePipeline",
+            "Flex2Pipeline",
             "FluxControlImg2ImgPipeline",
             "FluxControlInpaintPipeline",
             "FluxControlNetImg2ImgPipeline",
@@ -934,6 +935,7 @@
             EasyAnimateControlPipeline,
             EasyAnimateInpaintPipeline,
             EasyAnimatePipeline,
+            Flex2Pipeline,
             FluxControlImg2ImgPipeline,
             FluxControlInpaintPipeline,
             FluxControlNetImg2ImgPipeline,

src/diffusers/pipelines/__init__.py

@@ -127,6 +127,11 @@
         "AnimateDiffVideoToVideoPipeline",
         "AnimateDiffVideoToVideoControlNetPipeline",
     ]
+    _import_structure["flex"] = [
+        "Flex2Pipeline",
+        "Flex2PipelineOutput",
+    ]
+
     _import_structure["flux"] = [
         "FluxControlPipeline",
         "FluxControlInpaintPipeline",
@@ -572,6 +577,7 @@
             EasyAnimateInpaintPipeline,
             EasyAnimatePipeline,
         )
+        from .flex import Flex2Pipeline, Flex2PipelineOutput
         from .flux import (
             FluxControlImg2ImgPipeline,
             FluxControlInpaintPipeline,

src/diffusers/pipelines/flex/__init__.py

@@ -0,0 +1,23 @@
+from typing import TYPE_CHECKING
+
+from ...utils import (
+    DIFFUSERS_SLOW_IMPORT,
+    _LazyModule,
+)
+
+_import_structure = {"pipeline_flex": ["Flex2Pipeline"],
+                     "pipeline_output": ["Flex2PipelineOutput"]}
+
+if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
+    from .pipeline_flex import Flex2Pipeline
+    from .pipeline_output import Flex2PipelineOutput
+
+else:
+    import sys
+
+    sys.modules[__name__] = _LazyModule(
+        __name__,
+        globals()["__file__"],
+        _import_structure,
+        module_spec=__spec__,
+    )