diff --git a/people-and-planet-ai/weather-forecasting/serving/weather-model/pyproject.toml b/people-and-planet-ai/weather-forecasting/serving/weather-model/pyproject.toml index 43c03683ccd..be97b75e4e9 100644 --- a/people-and-planet-ai/weather-forecasting/serving/weather-model/pyproject.toml +++ b/people-and-planet-ai/weather-forecasting/serving/weather-model/pyproject.toml @@ -18,7 +18,7 @@ name = "weather-model" version = "1.0.0" dependencies = [ "datasets==4.0.0", - "torch==2.8.0", # make sure this matches the `container_uri` in `notebooks/3-training.ipynb` + "torch==2.12.0", # make sure this matches the `container_uri` in `notebooks/3-training.ipynb` "transformers==5.0.0", ]