Upload folder using huggingface_hub

block.py

@@ -0,0 +1,54 @@
+from typing import List
+import torch
+
+from diffusers.pipelines.modular_pipeline import PipelineState, PipelineBlock
+from diffusers.pipelines.modular_pipeline_utils import (
+    InputParam,
+    ComponentSpec,
+    OutputParam,
+)
+from image_gen_aux import DepthPreprocessor
+
+
+class DepthProcessorBlock(PipelineBlock):
+    @property
+    def expected_components(self):
+        return [
+            ComponentSpec(
+                name="depth_processor",
+                type_hint=DepthPreprocessor,
+                repo="depth-anything/Depth-Anything-V2-Large-hf",
+            )
+        ]
+
+    @property
+    def inputs(self) -> List[InputParam]:
+        return [
+            InputParam(
+                "image",
+                PipelineImageInput,
+                required=True,
+                description="Image(s) to use to extract depth maps",
+            )
+        ]
+
+    @property
+    def intermediates_outputs(self) -> List[OutputParam]:
+        return [
+            OutputParam(
+                "image",
+                type_hint=torch.Tensor,
+                description="Depth Map(s) of input Image(s)",
+            ),
+        ]
+
+    @torch.no_grad()
+    def __call__(self, pipeline, state: PipelineState) -> PipelineState:
+        block_state = self.get_block_state(state)
+
+        image = block_state.image
+        depth_map = pipeline.depth_processor(image)
+        block_state.image = depth_map.to(block_state.device)
+
+        self.add_block_state(state, block_state)
+        return pipeline, state

config.json

@@ -0,0 +1,21 @@
+{
+  "_class_name": "DepthProcessorBlock",
+  "auto_map": {
+    "PipelineBlock": "block.DepthProcessorBlock"
+  },
+  "_component_names": [
+    "depth_processor"
+  ],
+  "depth_processor": [
+    null,
+    null,
+    {
+      "repo": "depth-anything/Depth-Anything-V2-Large-hf",
+      "type_hint": [
+        "image_gen_aux",
+        "DepthPreprocessor"
+      ],
+      "variant": null
+    }
+  ]
+}

requirements.txt

@@ -0,0 +1 @@
+git+https://github.com/huggingface/image_gen_aux.git