update

Author: DN6

src/diffusers/__init__.py

@@ -399,6 +399,8 @@
 else:
     _import_structure["modular_pipelines"].extend(
         [
+            "Flux2AutoBlocks",
+            "Flux2ModularPipeline",
             "FluxAutoBlocks",
             "FluxKontextAutoBlocks",
             "FluxKontextModularPipeline",
@@ -1091,6 +1093,8 @@
         from .utils.dummy_torch_and_transformers_objects import *  # noqa F403
     else:
         from .modular_pipelines import (
+            Flux2AutoBlocks,
+            Flux2ModularPipeline,
             FluxAutoBlocks,
             FluxKontextAutoBlocks,
             FluxKontextModularPipeline,

src/diffusers/modular_pipelines/flux2/modular_blocks.py

@@ -204,7 +204,7 @@ def description(self):
 
 AUTO_BLOCKS = InsertableDict(
     [
-        ("text_encoder", Flux2TextEncoderStep()),
+        ("text_encoder", Flux2AutoTextEncoderStep()),
         ("image_encoder", Flux2AutoVaeEncoderStep()),
         ("denoise", Flux2CoreDenoiseStep()),
         ("decode", Flux2DecodeStep()),

src/diffusers/modular_pipelines/modular_pipeline.py

@@ -58,6 +58,7 @@
         ("wan", "WanModularPipeline"),
         ("flux", "FluxModularPipeline"),
         ("flux-kontext", "FluxKontextModularPipeline"),
+        ("flux2", "Flux2ModularPipeline"),
         ("qwenimage", "QwenImageModularPipeline"),
         ("qwenimage-edit", "QwenImageEditModularPipeline"),
         ("qwenimage-edit-plus", "QwenImageEditPlusModularPipeline"),