Skip to content

Commit 7a4e77c

Browse files
authored
Update TiRex model URL to avoid leakage (#249)
Updates TiRex model url with the diff model trained for ensuring non leakage. The underling LLM has also been trained with this TiRex version, to ensure no leakage.
1 parent b4ad1f4 commit 7a4e77c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

project/moirai-agent/gift_eval/tsf_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def make_forecast_batch(
135135
def get_tirex_forecast_fn(device="cuda"):
136136
from tirex import ForecastModel, load_model
137137

138-
model: ForecastModel = load_model("NX-AI/TiRex", device=device)
138+
model: ForecastModel = load_model("NX-AI/TiRex-1.1-gifteval", device=device)
139139
quantile_levels = ["0.1", "0.2", "0.3", "0.4", "0.5", "0.6", "0.7", "0.8", "0.9"]
140140

141141
def make_forecast_batch(

0 commit comments

Comments
 (0)