shchuro commited on
Commit
32f6d60
·
1 Parent(s): 0f4fb35

Add TiRex model

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. src/formatting.py +1 -0
app.py CHANGED
@@ -32,6 +32,7 @@ summary_urls = [
32
  "https://raw.githubusercontent.com/autogluon/fev/refs/heads/main/benchmarks/chronos_zeroshot/results/timesfm.csv",
33
  "https://raw.githubusercontent.com/autogluon/fev/refs/heads/main/benchmarks/chronos_zeroshot/results/timesfm-2.0.csv",
34
  "https://raw.githubusercontent.com/autogluon/fev/refs/heads/main/benchmarks/chronos_zeroshot/results/ttm-r2.csv",
 
35
  ]
36
 
37
  rename_cols = {
 
32
  "https://raw.githubusercontent.com/autogluon/fev/refs/heads/main/benchmarks/chronos_zeroshot/results/timesfm.csv",
33
  "https://raw.githubusercontent.com/autogluon/fev/refs/heads/main/benchmarks/chronos_zeroshot/results/timesfm-2.0.csv",
34
  "https://raw.githubusercontent.com/autogluon/fev/refs/heads/main/benchmarks/chronos_zeroshot/results/ttm-r2.csv",
35
+ "https://raw.githubusercontent.com/autogluon/fev/refs/heads/main/benchmarks/chronos_zeroshot/results/tirex.csv",
36
  ]
37
 
38
  rename_cols = {
src/formatting.py CHANGED
@@ -18,6 +18,7 @@ MODEL_URLS = {
18
  "timesfm": "google/timesfm-1.0-200m-pytorch",
19
  "timesfm-2.0": "google/timesfm-2.0-500m-pytorch",
20
  "ttm-r2": "ibm-granite/granite-timeseries-ttm-r2",
 
21
  }
22
 
23
 
 
18
  "timesfm": "google/timesfm-1.0-200m-pytorch",
19
  "timesfm-2.0": "google/timesfm-2.0-500m-pytorch",
20
  "ttm-r2": "ibm-granite/granite-timeseries-ttm-r2",
21
+ "tirex": "NX-AI/TiRex",
22
  }
23
 
24