Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move seed above optional params.

Browse Source
This commit is contained in:
Ryan Dick 2024-06-12 14:37:09 -04:00 committed by Kent Keirsey
parent 82a69a54ac
commit d661517d94
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/stable_diffusion/diffusers_pipeline.py
View File

@ -286,6 +286,7 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
scheduler_step_kwargs: dict[str, Any], scheduler_step_kwargs: dict[str, Any],
conditioning_data: TextConditioningData, conditioning_data: TextConditioningData,
noise: Optional[torch.Tensor], noise: Optional[torch.Tensor],
seed: int,
timesteps: torch.Tensor, timesteps: torch.Tensor,
init_timestep: torch.Tensor, init_timestep: torch.Tensor,
callback: Callable[[PipelineIntermediateState], None] = None, callback: Callable[[PipelineIntermediateState], None] = None,
@ -295,7 +296,6 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
mask: Optional[torch.Tensor] = None, mask: Optional[torch.Tensor] = None,
masked_latents: Optional[torch.Tensor] = None, masked_latents: Optional[torch.Tensor] = None,
gradient_mask: Optional[bool] = False, gradient_mask: Optional[bool] = False,
seed: int,
) -> torch.Tensor: ) -> torch.Tensor:
if init_timestep.shape[0] == 0: if init_timestep.shape[0] == 0:
return latents return latents