Skip to content

Commit 119c6f1

Browse files
chore(deps): bump torch from 2.4.0 to 2.8.0 in /people-and-planet-ai/weather-forecasting/serving/weather-model (#13554)
* chore(deps): bump torch Bumps [torch](https://github.com/pytorch/pytorch) from 2.4.0 to 2.8.0. - [Release notes](https://github.com/pytorch/pytorch/releases) - [Changelog](https://github.com/pytorch/pytorch/blob/main/RELEASE.md) - [Commits](pytorch/pytorch@v2.4.0...v2.8.0) --- updated-dependencies: - dependency-name: torch dependency-version: 2.8.0 dependency-type: direct:production ... Signed-off-by: dependabot[bot] <[email protected]> * update container_uri to match update --------- Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Katie McLaughlin <[email protected]>
1 parent ac9b6f1 commit 119c6f1

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

people-and-planet-ai/weather-forecasting/notebooks/3-training.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1381,7 +1381,7 @@
13811381
" display_name=\"weather-forecasting\",\n",
13821382
" python_package_gcs_uri=f\"gs://{bucket}/weather/weather-model-1.0.0.tar.gz\",\n",
13831383
" python_module_name=\"weather.trainer\",\n",
1384-
" container_uri=\"us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest\",\n",
1384+
" container_uri=\"us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-8.py310:latest\",\n",
13851385
")\n",
13861386
"job.run(\n",
13871387
" machine_type=\"n1-highmem-8\",\n",

people-and-planet-ai/weather-forecasting/serving/weather-model/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ name = "weather-model"
1818
version = "1.0.0"
1919
dependencies = [
2020
"datasets==4.0.0",
21-
"torch==2.4.0", # make sure this matches the `container_uri` in `notebooks/3-training.ipynb`
21+
"torch==2.8.0", # make sure this matches the `container_uri` in `notebooks/3-training.ipynb`
2222
"transformers==4.48.0",
2323
]
2424

0 commit comments

Comments
 (0)