diffusers-internal-dev
/

canny-filtering

Model card Files Files and versions

sayakpaul HF Staff commited on 6 days ago

Commit

17ad199

·

verified ·

1 Parent(s): 6d3538a

Update canny_block.py

Files changed (1) hide show

canny_block.py +2 -2

canny_block.py CHANGED Viewed

@@ -54,7 +54,7 @@ class CannyBlock(ModularPipelineBlocks):
     def intermediate_outputs(self) -> List[OutputParam]:
         return [
             OutputParam(
-                "canny_map",
                 type_hint=Image,
                 description="Canny map for input image",
             )
@@ -74,7 +74,7 @@ class CannyBlock(ModularPipelineBlocks):
     def __call__(self, components, state: PipelineState) -> PipelineState:
         block_state = self.get_block_state(state)
-        block_state.canny_map = self.compute_canny(
             components,
             block_state.image,
             block_state.low_threshold,

     def intermediate_outputs(self) -> List[OutputParam]:
         return [
             OutputParam(
+                "control_image",
                 type_hint=Image,
                 description="Canny map for input image",
             )
     def __call__(self, components, state: PipelineState) -> PipelineState:
         block_state = self.get_block_state(state)
+        block_state.control_image = self.compute_canny(
             components,
             block_state.image,
             block_state.low_threshold,