model/config.json

{
  "_class_name": "TrajectoryDiffusionModel",
  "_diffusers_version": "0.36.0",
  "hidden_dim": 256,
  "num_steps": 32,
  "time_embed_dim": 64
}