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",
|