Sampler › Denoiser › Time embedding
Expansion
×learned
Map the 768 encoding features to 1,536.
- in
- [B batch, 768 features]
- out
- [B batch, 1536 features]
- parameters
- 1,181,184
Parameters
1,181,184 parameters owned by this component
| Component | Calculation | Parameters |
|---|---|---|
| weights + bias | 1,181,184 | |
| Per-instance total | 1,181,184 | |
Across 1024 shared calls: parameters.
Derived from the pinned implementation’s constructors and configuration. The checkpoint’s tensors have not been counted. 306,011,162 of these parameters are trained; frozen GPT-2 and the decoder’s mask vector, 124,442,112, are not.
Sources2
Direct evidence
The denoiser owns the time embedding’s two affine layers, one rotary table for all blocks, an affine projection of the concatenated input, and the stack.
The implementation, pinned at revision e18c42acdfb7, not yet public
Direct evidence
The time embedding passes its MLP and a SiLU; the noised latents are concatenated with the previous clean estimate, zeros when there is none, and projected; the stack runs in bfloat16.
The implementation, pinned at revision e18c42acdfb7, not yet public