mean_flow_base_2(pretrained: bool = False, overrides: object = {})Base width, patch 2 — the smallest reported model.
Model Size
Parameters
pretrainedbool= FalseAccepted for signature parity with the rest of the zoo. No
MeanFlow checkpoint ships — the paper releases none, and a
True here would have nothing to load.**overridesobject= {}Optional
MeanFlowConfig field overrides.Returns
MeanFlowModelThe average-velocity network.
Notes
Reference: Geng et al., arXiv:2505.13447, 2025, Tables 2 and 4. Reports 6.17 FID at one function evaluation on ImageNet 256x256.
Examples
>>> from lucid.models import mean_flow_base_2
>>> config = mean_flow_base_2().config
>>> config.hidden_size, config.depth, config.num_patches
(768, 12, 256)