config.json

{
  "_class_name": "TemporalUnet",
  "_diffusers_version": "0.0.4",
  "horizon": 128,
  "dim": 32,
  "dim_mults": [1, 4, 8],
  "transition_dim": 14,
  "cond_dim": 3
}