mstar.model.pi05.components#
Pi0.5 components package — top-level classes live in
paligemma.py and action_expert.py. The shared transformer
building blocks (RMSNorm, GatedMLP, Attention, DecoderLayer) come from
mstar.model.components.
Modules
Flow-matching utilities for Pi0.5: timestep embedding, Euler step, state discretization. |
|
Tokenization wrapper for Pi0.5: PaliGemma tokenizer + state discretization. |