diff --git a/config/AEPSA/v2_PEMS-BAY.yaml b/config/AEPSA/v2_PEMS-BAY.yaml
index 7b5c97e..c40034d 100755
--- a/config/AEPSA/v2_PEMS-BAY.yaml
+++ b/config/AEPSA/v2_PEMS-BAY.yaml
@@ -19,9 +19,11 @@ data:
   val_ratio: 0.2
 
 model:
+  chebyshev_order: 3
   d_ff: 128
   d_model: 64
   dropout: 0.2
+  graph_hidden_dim: 32
   gpt_layers: 9
   gpt_path: ./GPT-2
   input_dim: 1