From b11ecc736a199fa2218eac693385ab437e3124b4 Mon Sep 17 00:00:00 2001
From: deepanker13 <deepanker.gupta@nutanix.com>
Date: Fri, 5 Jan 2024 15:11:00 +0530
Subject: [PATCH] torch version fix for python3.7

---
 sdk/python/kubeflow/storage_init_container/requirements.txt | 6 +++---
 sdk/python/setup.py                                         | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/sdk/python/kubeflow/storage_init_container/requirements.txt b/sdk/python/kubeflow/storage_init_container/requirements.txt
index cc63881807..951305587d 100644
--- a/sdk/python/kubeflow/storage_init_container/requirements.txt
+++ b/sdk/python/kubeflow/storage_init_container/requirements.txt
@@ -1,6 +1,6 @@
-torch>=1.8.2
-torchvision>=0.9.2
-torchaudio>=0.8.2
+torch>=1.8.1
+torchvision>=0.9.1
+torchaudio>=0.8.1
 einops==0.7.0
 transformers_stream_generator==0.0.4
 boto3==1.33.9
diff --git a/sdk/python/setup.py b/sdk/python/setup.py
index fb461153bc..ff35a334bf 100644
--- a/sdk/python/setup.py
+++ b/sdk/python/setup.py
@@ -31,9 +31,9 @@
     "retrying>=1.3.3",
     "boto3>=1.33.9",
     "transformers>=4.20.0",
-    "torch>=1.8.2",
-    "torchvision>=0.9.2",
-    "torchaudio>=0.8.2",
+    "torch>=1.8.1",
+    "torchvision>=0.9.1",
+    "torchaudio>=0.8.1",
     "einops>=0.7.0",
     "transformers_stream_generator>=0.0.4",
     "peft>=0.7.0",