config.yaml

model:
    target: pixelflow.model.PixelFlowModel
    params:
        num_attention_heads: 16
        attention_head_dim: 72
        in_channels: 3
        out_channels: 3
        depth: 28
        num_classes: 0
        patch_size: 4
        attention_bias: true
        cross_attention_dim: 2048
scheduler:
  num_train_timesteps: 1000
  num_stages: 4
  pyramid_shift: false
seed: 42