config.json

{
  "_class_name": "TemporalUnet",
  "_diffusers_version": "0.0.4",
  "training_horizon": 512,
  "dim": 32,
  "dim_mults": [1, 4, 8],
  "predict_epsilon": false,
  "clip_denoised": true,  
  "transition_dim": 14,
  "cond_dim": 3
}