From f103011dbb775b5ab269f4445520a7101b71af47 Mon Sep 17 00:00:00 2001
From: Mihir Patel <mihir.v.patel7@gmail.com>
Date: Mon, 26 Feb 2024 17:10:43 -0500
Subject: [PATCH] fix torchvision

---
 docker/generate_build_matrix.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/generate_build_matrix.py b/docker/generate_build_matrix.py
index 5bcada23ea..215acef5b8 100644
--- a/docker/generate_build_matrix.py
+++ b/docker/generate_build_matrix.py
@@ -24,7 +24,7 @@
 
 def _get_torchvision_version(pytorch_version: str):
     if pytorch_version == '2.2.1':
-        return '0.17.0'
+        return '0.17.1'
     if pytorch_version == '2.1.2':
         return '0.16.2'
     if pytorch_version == '2.0.1':