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 b15bface40..f93be68b97 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.4.0", # make sure this matches the `container_uri` in `notebooks/3-training.ipynb` + "torch==2.13.0", # make sure this matches the `container_uri` in `notebooks/3-training.ipynb` "transformers==5.13.0", ]