rectified_flow_bedroom_gen

→RectifiedFlowForImageGeneration

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

source edit

Rectified Flow generator for LSUN Bedroom 256 x 256.

Model Size

Parameters

pretrainedbool= False

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

**overridesobject= {}

Optional RectifiedFlowConfig field overrides.

Returns

RectifiedFlowForImageGeneration

Generator configured for LSUN Bedroom.

Notes

Reference: Liu, Gong, and Liu, "Flow Straight and Fast", ICLR, 2023 (arXiv:2209.03003).

Examples

>>> from lucid.models.generative.rectified_flow import (
...     rectified_flow_bedroom_gen,
... )
>>> rectified_flow_bedroom_gen().rectified_flow.input_dim
196608

Used by 1

lucid.models.generative.rectified_flow