From d780d1a42a855d21cc8cd7cfbfa1287f846c44ba Mon Sep 17 00:00:00 2001
From: DN6 <dhruv.nair@gmail.com>
Date: Fri, 28 Nov 2025 15:27:30 +0530
Subject: [PATCH] update

---
 src/diffusers/__init__.py                               | 4 ++++
 src/diffusers/modular_pipelines/flux2/modular_blocks.py | 2 +-
 src/diffusers/modular_pipelines/modular_pipeline.py     | 1 +
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py
index 8a81beca97..6c0cebc8de 100644
--- a/src/diffusers/__init__.py
+++ b/src/diffusers/__init__.py
@@ -399,6 +399,8 @@ except OptionalDependencyNotAvailable:
 else:
     _import_structure["modular_pipelines"].extend(
         [
+            "Flux2AutoBlocks",
+            "Flux2ModularPipeline",
             "FluxAutoBlocks",
             "FluxKontextAutoBlocks",
             "FluxKontextModularPipeline",
@@ -1091,6 +1093,8 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
         from .utils.dummy_torch_and_transformers_objects import *  # noqa F403
     else:
         from .modular_pipelines import (
+            Flux2AutoBlocks,
+            Flux2ModularPipeline,
             FluxAutoBlocks,
             FluxKontextAutoBlocks,
             FluxKontextModularPipeline,
diff --git a/src/diffusers/modular_pipelines/flux2/modular_blocks.py b/src/diffusers/modular_pipelines/flux2/modular_blocks.py
index a17f413667..e601829d08 100644
--- a/src/diffusers/modular_pipelines/flux2/modular_blocks.py
+++ b/src/diffusers/modular_pipelines/flux2/modular_blocks.py
@@ -204,7 +204,7 @@ class Flux2CoreDenoiseStep(SequentialPipelineBlocks):
 
 AUTO_BLOCKS = InsertableDict(
     [
-        ("text_encoder", Flux2TextEncoderStep()),
+        ("text_encoder", Flux2AutoTextEncoderStep()),
         ("image_encoder", Flux2AutoVaeEncoderStep()),
         ("denoise", Flux2CoreDenoiseStep()),
         ("decode", Flux2DecodeStep()),
diff --git a/src/diffusers/modular_pipelines/modular_pipeline.py b/src/diffusers/modular_pipelines/modular_pipeline.py
index c7285e38fd..38f447b0cf 100644
--- a/src/diffusers/modular_pipelines/modular_pipeline.py
+++ b/src/diffusers/modular_pipelines/modular_pipeline.py
@@ -58,6 +58,7 @@ MODULAR_PIPELINE_MAPPING = OrderedDict(
         ("wan", "WanModularPipeline"),
         ("flux", "FluxModularPipeline"),
         ("flux-kontext", "FluxKontextModularPipeline"),
+        ("flux2", "Flux2ModularPipeline"),
         ("qwenimage", "QwenImageModularPipeline"),
         ("qwenimage-edit", "QwenImageEditModularPipeline"),
         ("qwenimage-edit-plus", "QwenImageEditPlusModularPipeline"),