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