Update block.py
Browse files
block.py
CHANGED
@@ -76,7 +76,7 @@ class SDXLDiffDiffPrepareLatentsStep(PipelineBlock):
|
|
76 |
]
|
77 |
|
78 |
@property
|
79 |
-
def
|
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
|
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
|
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",
|