rectified_flow_celeba_hq

→RectifiedFlowModel

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

source edit

Rectified Flow velocity field for CelebA-HQ 256 x 256.

Model Size

Parameters

pretrainedbool= False

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

**overridesobject= {}

Optional RectifiedFlowConfig field overrides.

Returns

RectifiedFlowModel

Velocity field with filtered resampling and a resolution pyramid.

Notes

Reference: Liu, Gong, and Liu, "Flow Straight and Fast", ICLR, 2023 (arXiv:2209.03003). This is one of the datasets the paper's transfer experiments run on — the method is not restricted to a Gaussian source, and the same objective transports between two image domains.

Examples

>>> from lucid.models.generative.rectified_flow import rectified_flow_celeba_hq
>>> rectified_flow_celeba_hq().config.embedding_type
'fourier'

Used by 1

lucid.models.generative.rectified_flow