This repository was archived by the owner on Feb 7, 2025. It is now read-only.
This repository was archived by the owner on Feb 7, 2025. It is now read-only.
Allow for dropout on cross-attention blocks of diffusion model #406
Description
In some cases, having dropout helps extrapolating the conditioning. Currently, it's set to zero in the middle of the initialisation chain. It'll be easy to allow for dropout to be specified.
Metadata
Metadata
Assignees
Labels
No labels