| { |
| "model_type": "flow_match_relay", |
| "architectures": [ |
| "FlowMatchRelayModel" |
| ], |
| "auto_map": { |
| "AutoConfig": "configuration_flow_match.FlowMatchRelayConfig", |
| "AutoModel": "modeling_flow_match.FlowMatchRelayModel" |
| }, |
| "in_channels": 3, |
| "base_channels": 64, |
| "channel_mults": [ |
| 1, |
| 2, |
| 4 |
| ], |
| "n_classes": 10, |
| "cond_dim": 256, |
| "use_relay": true, |
| "relay_patch_dim": 16, |
| "relay_n_anchors": 16, |
| "relay_n_phases": 3, |
| "relay_pw_hidden": 32, |
| "relay_gate_init": -3.0, |
| "relay_mode": "channel", |
| "image_size": 32, |
| "n_sample_steps": 50, |
| "torch_dtype": "float32" |
| } |
