From 04de34b82ef56aa41c44b88953142b623a67ed47 Mon Sep 17 00:00:00 2001
From: DN6 <dhruv.nair@gmail.com>
Date: Wed, 10 Dec 2025 11:18:10 +0530
Subject: [PATCH] update

---
 .../modular_pipelines/flux2/__init__.py       |  8 ++---
 .../dummy_torch_and_transformers_objects.py   | 30 +++++++++++++++++++
 2 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/src/diffusers/modular_pipelines/flux2/__init__.py b/src/diffusers/modular_pipelines/flux2/__init__.py
index 0af7819dd0..3bc4587054 100644
--- a/src/diffusers/modular_pipelines/flux2/__init__.py
+++ b/src/diffusers/modular_pipelines/flux2/__init__.py
@@ -24,7 +24,6 @@ else:
     _import_structure["encoders"] = [
         "Flux2TextEncoderStep",
         "Flux2RemoteTextEncoderStep",
-        "Flux2ProcessImagesInputStep",
         "Flux2VaeEncoderStep",
     ]
     _import_structure["before_denoise"] = [
@@ -41,12 +40,13 @@ else:
     ]
     _import_structure["decoders"] = ["Flux2DecodeStep"]
     _import_structure["inputs"] = [
+        "Flux2ProcessImagesInputStep",
         "Flux2TextInputStep",
-        "Flux2ImageInputStep",
     ]
     _import_structure["modular_blocks"] = [
         "ALL_BLOCKS",
         "AUTO_BLOCKS",
+        "REMOTE_AUTO_BLOCKS",
         "TEXT2IMAGE_BLOCKS",
         "IMAGE_CONDITIONED_BLOCKS",
         "Flux2AutoBeforeDenoiseStep",
@@ -83,19 +83,19 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
             Flux2LoopDenoiser,
         )
         from .encoders import (
-            Flux2ProcessImagesInputStep,
             Flux2RemoteTextEncoderStep,
             Flux2TextEncoderStep,
             Flux2VaeEncoderStep,
         )
         from .inputs import (
-            Flux2ImageInputStep,
+            Flux2ProcessImagesInputStep,
             Flux2TextInputStep,
         )
         from .modular_blocks import (
             ALL_BLOCKS,
             AUTO_BLOCKS,
             IMAGE_CONDITIONED_BLOCKS,
+            REMOTE_AUTO_BLOCKS,
             TEXT2IMAGE_BLOCKS,
             Flux2AutoBeforeDenoiseStep,
             Flux2AutoBlocks,
diff --git a/src/diffusers/utils/dummy_torch_and_transformers_objects.py b/src/diffusers/utils/dummy_torch_and_transformers_objects.py
index e6cf26a125..c81f4ab329 100644
--- a/src/diffusers/utils/dummy_torch_and_transformers_objects.py
+++ b/src/diffusers/utils/dummy_torch_and_transformers_objects.py
@@ -2,6 +2,36 @@
 from ..utils import DummyObject, requires_backends
 
 
+class Flux2AutoBlocks(metaclass=DummyObject):
+    _backends = ["torch", "transformers"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch", "transformers"])
+
+    @classmethod
+    def from_config(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "transformers"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "transformers"])
+
+
+class Flux2ModularPipeline(metaclass=DummyObject):
+    _backends = ["torch", "transformers"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch", "transformers"])
+
+    @classmethod
+    def from_config(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "transformers"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["torch", "transformers"])
+
+
 class FluxAutoBlocks(metaclass=DummyObject):
     _backends = ["torch", "transformers"]