config.json

{
  "d_model": 256,
  "num_heads": 4,
  "num_layers": 6,
  "ff_dim": 1024,
  "patch_size": 32,
  "context_length": 1024,
  "max_prediction_length": 512,
  "num_quantiles": 9,
  "num_predict_tokens": 2,
  "dropout": 0.0,
  "quantile_levels": [
    0.1,
    0.2,
    0.3,
    0.4,
    0.5,
    0.6,
    0.7,
    0.8,
    0.9
  ],
  "max_patches": 32
}