config.json

{
  "etas_end": 0.99,
  "flow_kwargs": {
    "pretrained_path": null
  },
  "flow_model": "raft",
  "kappa": 2.0,
  "min_noise_level": 0.04,
  "p": 0.3,
  "synthesis_kwargs": {
    "channels": [
      128,
      256,
      384,
      512
    ],
    "dropout": 0.0,
    "expansion_rate": 1.5,
    "grid_attn": true,
    "heads": 1,
    "in_channels": 3,
    "num_conv_blocks": 1,
    "temb_channels": 512,
    "window_attn": true,
    "window_size": 8
  },
  "timesteps": 20,
  "warping_kwargs": {
    "channels": [
      128,
      256,
      384,
      512
    ],
    "in_channels": 3
  }
}