integration test fix for python3.7

deepanker13 · Jan 5, 2024 · 937e85a · 937e85a
1 parent 4c65a5a
commit 937e85a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/sdk/python/setup.py b/sdk/python/setup.py
@@ -30,10 +30,10 @@
     "kubernetes>=23.6.0",
     "retrying>=1.3.3",
     "boto3>=1.33.9",
-    "transformers>=4.35.2",
-    "torch>=2.1.1",
-    "torchvision>=0.16.1",
-    "torchaudio>=2.1.1",
+    "transformers>=4.20.0",
+    "torch>=1.8.2",
+    "torchvision>=0.9.2",
+    "torchaudio>=0.8.2",
     "einops>=0.7.0",
     "transformers_stream_generator>=0.0.4",
     "peft>=0.7.0",