YiYiXu commited on
Commit
cda6908
·
verified ·
1 Parent(s): ee4b271

Update block.py

Browse files
Files changed (1) hide show
  1. block.py +3 -3
block.py CHANGED
@@ -76,7 +76,7 @@ class SDXLDiffDiffPrepareLatentsStep(PipelineBlock):
76
  ]
77
 
78
  @property
79
- def intermediates_inputs(self) -> List[InputParam]:
80
  return [
81
  InputParam("generator"),
82
  InputParam("timesteps",type_hint=torch.Tensor, description="The timesteps to use for sampling. Can be generated in set_timesteps step."),
@@ -86,7 +86,7 @@ class SDXLDiffDiffPrepareLatentsStep(PipelineBlock):
86
  ]
87
 
88
  @property
89
- def intermediates_outputs(self) -> List[OutputParam]:
90
  return [
91
  OutputParam("latents", type_hint=torch.Tensor, description="The initial latents to use for the denoising process"),
92
  OutputParam("original_latents", type_hint=torch.Tensor, description="The initial latents to use for the denoising process"),
@@ -150,7 +150,7 @@ class SDXLDiffDiffLoopBeforeDenoiser(PipelineBlock):
150
  ]
151
 
152
  @property
153
- def intermediates_inputs(self) -> List[str]:
154
  return [
155
  InputParam(
156
  "latents",
 
76
  ]
77
 
78
  @property
79
+ def intermediate_inputs(self) -> List[InputParam]:
80
  return [
81
  InputParam("generator"),
82
  InputParam("timesteps",type_hint=torch.Tensor, description="The timesteps to use for sampling. Can be generated in set_timesteps step."),
 
86
  ]
87
 
88
  @property
89
+ def intermediate_outputs(self) -> List[OutputParam]:
90
  return [
91
  OutputParam("latents", type_hint=torch.Tensor, description="The initial latents to use for the denoising process"),
92
  OutputParam("original_latents", type_hint=torch.Tensor, description="The initial latents to use for the denoising process"),
 
150
  ]
151
 
152
  @property
153
+ def intermediate_inputs(self) -> List[str]:
154
  return [
155
  InputParam(
156
  "latents",