YiYiXu
/

modular-diffdiff-0704

Model card Files Files and versions

YiYiXu HF Staff commited on May 21, 2025

Commit

d738856

·

verified ·

1 Parent(s): 34e7c7d

Update block.py

Files changed (1) hide show

block.py +2 -2

block.py CHANGED Viewed

@@ -9,7 +9,7 @@ from diffusers.modular_pipelines import (
     ComponentSpec,
     AutoPipelineBlocks
 )
-from diffusers.image_processor import VaeImageProcessor
 from diffusers.schedulers import EulerDiscreteScheduler
 from diffusers.models import AutoencoderKL
 from diffusers.configuration_utils import FrozenDict
@@ -54,7 +54,7 @@ class SDXLDiffDiffPrepareLatentsStep(PipelineBlock):
     @property
     def inputs(self) -> List[Tuple[str, Any]]:
         return [
-            InputParam("diffdiff_map",required=True),
             InputParam(
                 "latents",
                 type_hint=Optional[torch.Tensor],

     ComponentSpec,
     AutoPipelineBlocks
 )
+from diffusers.image_processor import VaeImageProcessor, PipelineImageInput
 from diffusers.schedulers import EulerDiscreteScheduler
 from diffusers.models import AutoencoderKL
 from diffusers.configuration_utils import FrozenDict
     @property
     def inputs(self) -> List[Tuple[str, Any]]:
         return [
+            InputParam("diffdiff_map",type_hint=PipelineImageInput, required=True),
             InputParam(
                 "latents",
                 type_hint=Optional[torch.Tensor],