flow_matching_imagenet128_gen

→FlowMatchingForImageGeneration

flow_matching_imagenet128_gen(pretrained: bool = False, overrides: object = {})

source edit

Flow Matching generator for ImageNet 128 x 128.

Model Size

Parameters

pretrainedbool= False

Reserved for future weight registration; currently a no-op.

**overridesobject= {}

Optional FlowMatchingConfig field overrides.

Returns

FlowMatchingForImageGeneration

Generator configured for ImageNet at 128 x 128.

Notes

Reference: Lipman et al., "Flow Matching for Generative Modeling", ICLR, 2023 (arXiv:2210.02747). Reported: 20.9 FID / 2.90 bits/dim.

Examples

>>> from lucid.models.generative.flow_matching import (
...     flow_matching_imagenet128_gen,
... )
>>> flow_matching_imagenet128_gen().flow_matching.config.sample_size
128

Used by 1

lucid.models.generative.flow_matching