Skip to content

constants

constants

AttentionDecompositionMode

Bases: StrEnum

Attention computation strategies for DFormer.

Axis

Bases: StrEnum

Spatial image axes.

PositionalEncodingType

Bases: StrEnum

Types of positional encodings.

AttentionType

Bases: StrEnum

Types of attention mechanisms.

ConditioningType

Bases: StrEnum

Types of conditional modulation for transformers.