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

Separate inputs in denoise context

Browse Source
This commit is contained in:
Sergey Borisov
2024-07-16 19:30:29 +03:00
parent 9f088d1bf5
commit 608cbe3f5c
4 changed files with 33 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
invokeai/backend/stable_diffusion/denoise_context.py
View File

@ -30,8 +30,8 @@ class UNetKwargs:
@dataclass
class DenoiseContext:
latents: torch.Tensor
class DenoiseInputs:
orig_latents: torch.Tensor
scheduler_step_kwargs: dict[str, Any]
conditioning_data: TextConditioningData
noise: Optional[torch.Tensor]
@ -39,10 +39,15 @@ class DenoiseContext:
timesteps: torch.Tensor
init_timestep: torch.Tensor
@dataclass
class DenoiseContext:
inputs: DenoiseInputs
scheduler: SchedulerMixin
unet: Optional[UNet2DConditionModel] = None
orig_latents: Optional[torch.Tensor] = None
latents: Optional[torch.Tensor] = None
step_index: Optional[int] = None
timestep: Optional[torch.Tensor] = None
unet_kwargs: Optional[UNetKwargs] = None