YiYiXu
/

modular-depth-block

Model card Files Files and versions

YiYiXu HF Staff commited on May 15, 2025

Commit

462eb62

·

verified ·

1 Parent(s): 66b6384

Update block.py

Files changed (1) hide show

block.py +1 -1

block.py CHANGED Viewed

@@ -58,7 +58,7 @@ class DepthProcessorBlock(PipelineBlock):
         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)

         block_state = self.get_block_state(state)
         image = block_state.image
+        depth_map = pipeline.depth_processor(image, return_type="pt")
         block_state.image = depth_map.to(block_state.device)
         self.add_block_state(state, block_state)