Skip to content
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

@virginiafdez

Description

@virginiafdez

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
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions