Torchebm > Models > Components > Transformer

Contents

Classes

• AdaLNZeroBlock - Transformer block with adaLN-Zero conditioning.
• FeedForward - No description available.
• MultiheadSelfAttention - Self-attention wrapper with batch-first API.

Functions

• modulate() - No description available.

API Reference

torchebm.models.components.transformer