Skip to content

Commit f2d5bd3

Browse files
authored
update TiRex model url
1 parent 9a78b80 commit f2d5bd3

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)