model/config.json

{
  "_class_name": "GPSDynamicsModel",
  "_diffusers_version": "0.37.0",
  "action_dim": 3,
  "dropout": 0.1,
  "edge_feature_dim": 4,
  "hidden_dim": 256,
  "max_num_nodes": 160,
  "num_embeds_ada_norm": 1000,
  "num_heads": 8,
  "num_layers": 8,
  "num_next_frames": 5,
  "num_prev_frames": 3
}