Update block.py

block.py

@@ -9,7 +9,7 @@ from diffusers.modular_pipelines import (
     ComponentSpec, 
     AutoPipelineBlocks
 )
-from diffusers.image_processor import VaeImageProcessor
+from diffusers.image_processor import VaeImageProcessor, PipelineImageInput
 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("diffdiff_map",type_hint=PipelineImageInput, required=True),
             InputParam(
                 "latents", 
                 type_hint=Optional[torch.Tensor],