Skip to content

Commit d136ae3

Browse files
committed
update input for loop blocks, do not need to include intermediate
1 parent 1b89ac1 commit d136ae3

File tree

1 file changed

+0
-28
lines changed
  • src/diffusers/modular_pipelines/stable_diffusion_xl

1 file changed

+0
-28
lines changed

src/diffusers/modular_pipelines/stable_diffusion_xl/denoise.py

Lines changed: 0 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -68,18 +68,11 @@ def intermediates_inputs(self) -> List[str]:
6868
),
6969
]
7070

71-
@property
72-
def intermediates_outputs(self) -> List[OutputParam]:
73-
return [OutputParam("scaled_latents", type_hint=torch.Tensor, description="The scaled latents input for denoiser")]
74-
75-
76-
7771
@torch.no_grad()
7872
def __call__(self, components: StableDiffusionXLModularLoader, block_state: BlockState, i: int, t: int):
7973

8074
block_state.scaled_latents = components.scheduler.scale_model_input(block_state.latents, t)
8175

82-
8376
return components, block_state
8477

8578
# loop step (1): prepare latent input for denoiser (with inpainting)
@@ -120,9 +113,6 @@ def intermediates_inputs(self) -> List[str]:
120113
),
121114
]
122115

123-
@property
124-
def intermediates_outputs(self) -> List[OutputParam]:
125-
return [OutputParam("scaled_latents", type_hint=torch.Tensor, description="The scaled latents input for denoiser")]
126116

127117
@staticmethod
128118
def check_inputs(components, block_state):
@@ -187,12 +177,6 @@ def inputs(self) -> List[Tuple[str, Any]]:
187177
@property
188178
def intermediates_inputs(self) -> List[str]:
189179
return [
190-
InputParam(
191-
"scaled_latents",
192-
required=True,
193-
type_hint=torch.Tensor,
194-
description="The prepared latents input to use for the denoiser. Can be generated in latent step within the denoise loop."
195-
),
196180
InputParam(
197181
"num_inference_steps",
198182
required=True,
@@ -319,12 +303,6 @@ def intermediates_inputs(self) -> List[str]:
319303
type_hint=List[float],
320304
description="The controlnet keep values to use for the denoising process. Can be generated in prepare_controlnet_inputs step."
321305
),
322-
InputParam(
323-
"scaled_latents",
324-
required=True,
325-
type_hint=torch.Tensor,
326-
description="The prepared latents input to use for the denoiser. Can be generated in latent step within the denoise loop."
327-
),
328306
InputParam(
329307
"timestep_cond",
330308
type_hint=Optional[torch.Tensor],
@@ -492,12 +470,6 @@ def inputs(self) -> List[Tuple[str, Any]]:
492470
def intermediates_inputs(self) -> List[str]:
493471
return [
494472
InputParam("generator"),
495-
InputParam(
496-
"latents",
497-
required=True,
498-
type_hint=torch.Tensor,
499-
description="The initial latents to use for the denoising process. Can be generated in prepare_latent step."
500-
),
501473
]
502474

503475
@property

0 commit comments

Comments
 (0)