rectified_flow_afhq_cat(pretrained: bool = False, overrides: object = {})Rectified Flow velocity field for AFHQ-Cat 256 x 256.
Model Size
Parameters
pretrainedbool= FalseReserved for future weight registration; currently a no-op.
**overridesobject= {}Optional
RectifiedFlowConfig field overrides.Returns
RectifiedFlowModelVelocity field with filtered resampling and a resolution pyramid.
Notes
Reference: Liu, Gong, and Liu, "Flow Straight and Fast", ICLR, 2023 (arXiv:2209.03003). The paper's Figure 1 uses this dataset for both generation and face-to-face transfer.
Examples
>>> from lucid.models.generative.rectified_flow import rectified_flow_afhq_cat
>>> rectified_flow_afhq_cat().config.num_res_blocks
2