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.