| 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 | |