update

Author: DN6

src/diffusers/modular_pipelines/flux2/__init__.py

@@ -24,7 +24,6 @@
     _import_structure["encoders"] = [
         "Flux2TextEncoderStep",
         "Flux2RemoteTextEncoderStep",
-        "Flux2ProcessImagesInputStep",
         "Flux2VaeEncoderStep",
     ]
     _import_structure["before_denoise"] = [
@@ -41,12 +40,13 @@
     ]
     _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 @@
             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,

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"]