mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
16 lines
549 B
Python
16 lines
549 B
Python
import torch
|
|
|
|
|
|
def merge_intermediate_latents_with_init_latents(
|
|
init_latents: torch.Tensor,
|
|
intermediate_latents: torch.Tensor,
|
|
timestep: float,
|
|
noise: torch.Tensor,
|
|
inpaint_mask: torch.Tensor,
|
|
) -> torch.Tensor:
|
|
# Noise the init_latents for the current timestep.
|
|
noised_init_latents = noise * timestep + (1.0 - timestep) * init_latents
|
|
|
|
# Merge the intermediate_latents with the noised_init_latents using the inpaint_mask.
|
|
return intermediate_latents * inpaint_mask + noised_init_latents * (1.0 - inpaint_mask)
|