config.json

{
  "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"
}