meta') and loaded to GPU only when their specific submodule has its forward` method called.

noise_image_embeddings

<
source
>
(
image_embeds: Tensor
noise_level: int
noise: typing.Optional[torch.FloatTensor] = None
generator: typing.Optional[torch._C.Generator] = None

)



Add noise to the image embeddings. The amount of noise is controlled by a noise_level input. A higher
noise_level increases the variance in the final un-noised images.
The