config.json

{
  "_class_name": "DualEncoderEpsNetwork",
  "_diffusers_version": "0.0.4",
  "type": "diffusion",
  "network": "dualenc",
  "hidden_dim": 128,
  "num_convs": 6,
  "num_convs_local": 4,
  "cutoff": 10.0,
  "mlp_act": "relu",
  "beta_schedule": "sigmoid",
  "beta_start": 1e-7,
  "beta_end": 2e-3,
  "num_diffusion_timesteps": 5000,
  "edge_order": 3,
  "edge_encoder": "mlp",
  "smooth_conv": true
}