From 3587902a794283c25a33bae45f86eaa120df872a Mon Sep 17 00:00:00 2001 From: Oleksandr Shchur Date: Wed, 19 Aug 2026 13:16:08 +0000 Subject: [PATCH] tirex-2: cache checkpoint from S3 in wrapper Wrap model_path with fev.utils.maybe_cache_from_s3 so the tirex-2 wrapper accepts an s3:// prefix (matching the chronos/toto wrappers), downloading the checkpoint locally before load_model. --- models/tirex-2/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/tirex-2/model.py b/models/tirex-2/model.py index d14c9ab..ffb947f 100644 --- a/models/tirex-2/model.py +++ b/models/tirex-2/model.py @@ -32,7 +32,7 @@ def __init__( def _load_model(self) -> ForecastModel: if self._model is None: load_device = "cuda" if self.device.startswith("cuda") else self.device - self._model = load_model(self.model_path, device=load_device) + self._model = load_model(fev.utils.maybe_cache_from_s3(self.model_path), device=load_device) self._model.model.to(self.device) return self._model