flow_matching_imagenet32_gen

→FlowMatchingForImageGeneration

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

source edit

Flow Matching generator for ImageNet 32 x 32.

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 32 x 32.

Notes

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

Examples

>>> from lucid.models.generative.flow_matching import (
...     flow_matching_imagenet32_gen,
... )
>>> flow_matching_imagenet32_gen().flow_matching.path
'ot'

Used by 1

lucid.models.generative.flow_matching