removing torch ad dependency

deepanker13 · Jan 7, 2024 · ec461a1 · ec461a1
1 parent 9739d53
commit ec461a1
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 7 deletions.
diff --git 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
@@ -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"],
 )