From 937e85aae62c0c62b3f4f15bee8cb6e5dc2fcaae Mon Sep 17 00:00:00 2001 From: deepanker13 Date: Fri, 5 Jan 2024 14:31:57 +0530 Subject: [PATCH] integration test fix for python3.7 --- sdk/python/setup.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/sdk/python/setup.py b/sdk/python/setup.py index ae4fe6290a..fb461153bc 100644 --- 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",