mean_flow_medium_2(pretrained: bool = False, overrides: object = {})Medium width, patch 2.
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 5.01 FID at one function evaluation.
A size DiT does not define — the paper introduces it to fill the gap between Base and Large, at Large's width and two thirds of its depth.
Examples
>>> from lucid.models import mean_flow_medium_2
>>> config = mean_flow_medium_2().config
>>> config.hidden_size, config.depth
(1024, 16)