From ec461a14a4b8a56e2b9db94b21bad1a946eaa690 Mon Sep 17 00:00:00 2001
From: deepanker13 <deepanker.gupta@nutanix.com>
Date: Fri, 5 Jan 2024 19:01:53 +0530
Subject: [PATCH] removing torch ad dependency

---
 sdk/python/kubeflow/storage_init_container/requirements.txt | 3 ---
 sdk/python/setup.py                                         | 4 ----
 2 files changed, 7 deletions(-)

diff --git a/sdk/python/kubeflow/storage_init_container/requirements.txt b/sdk/python/kubeflow/storage_init_container/requirements.txt
index 500f61ec02..d82b00d972 100644
--- a/sdk/python/kubeflow/storage_init_container/requirements.txt
+++ b/sdk/python/kubeflow/storage_init_container/requirements.txt
@@ -1,6 +1,3 @@
-torch>=1.13.1
-torchvision>=0.9.1
-torchaudio>=0.8.1
 einops>=0.6.1
 transformers_stream_generator==0.0.4
 boto3==1.33.9
diff --git a/sdk/python/setup.py b/sdk/python/setup.py
index 9a9a30e245..0255d42137 100644
--- a/sdk/python/setup.py
+++ b/sdk/python/setup.py
@@ -31,9 +31,6 @@
     "retrying>=1.3.3",
     "boto3>=1.33.9",
     "transformers>=4.20.0",
-    "torch>=1.13.1",
-    "torchvision>=0.9.1",
-    "torchaudio>=0.8.1",
     "einops>=0.6.1",
     "transformers_stream_generator>=0.0.4",
     "peft>=0.3.0",
@@ -75,5 +72,4 @@
     install_requires=REQUIRES,
     tests_require=TESTS_REQUIRES,
     extras_require={"test": TESTS_REQUIRES},
-    dependency_links=["https://download.pytorch.org/whl/cu118"],
 )