diff --git a/dataflow/run-inference/requirements.txt b/dataflow/run-inference/requirements.txt index d0376ac202e..ee5714a2b3e 100644 --- a/dataflow/run-inference/requirements.txt +++ b/dataflow/run-inference/requirements.txt @@ -1,3 +1,3 @@ apache-beam[gcp]==2.49.0 torch==2.2.2 -transformers==5.0.0rc3 +transformers==5.3.0 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..6820345772e 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 @@ -19,7 +19,7 @@ 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` - "transformers==5.0.0", + "transformers==5.3.0", ] [project.scripts]