config.json

{
  "in_channels": 1,
  "out_channels": 1,
  "time_embedding_dim": 32
}