model/config.json

{
  "_class_name": "GNSDynamicsModel",
  "_diffusers_version": "0.37.0",
  "dt": 0.1,
  "edge_input_size": 4,
  "hidden_size": 128,
  "input_sequence_length": 1,
  "message_passing_num": 15,
  "nnode_types": 2,
  "node_input_size": 5,
  "position_noise_std": 0.0001,
  "velocity_noise_std": 0.0
}