Bump the python group with 2 updates (#1509)

Bumps the python group with 2 updates: torch and torchvision.


Updates `torch` from 2.1.2 to 2.2.1+cu121

Updates `torchvision` from 0.16.2 to 0.17.1+cu121

---
updated-dependencies:
- dependency-name: torch
  dependency-type: direct:production
  update-type: version-update:semver-minor
  dependency-group: python
- dependency-name: torchvision
  dependency-type: direct:production
  update-type: version-update:semver-minor
  dependency-group: python
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

Author: dependabot[bot]

requirements-torch-cuda.txt

@@ -4,8 +4,8 @@ tensorflow-text~=2.16.1
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.1.2
-torchvision==0.16.2
+torch==2.2.1+cu121
+torchvision==0.17.1+cu121
 
 # Jax cpu-only version.
 jax[cpu]