fixing library versions for python3.7

deepanker13 · Jan 5, 2024 · 257a535 · 257a535
1 parent 580b795
commit 257a535
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/sdk/python/kubeflow/storage_init_container/requirements.txt b/sdk/python/kubeflow/storage_init_container/requirements.txt
@@ -1,9 +1,9 @@
 torch>=1.8.1
 torchvision>=0.9.1
 torchaudio>=0.8.1
-einops==0.7.0
+einops>=0.6.1
 transformers_stream_generator==0.0.4
 boto3==1.33.9
 transformers>=4.20.0
-peft>=0.7.0
-huggingface_hub==0.19.4
+peft>=0.3.0
+huggingface_hub==0.16.4
diff --git a/sdk/python/setup.py b/sdk/python/setup.py
@@ -34,11 +34,11 @@
     "torch>=1.8.1",
     "torchvision>=0.9.1",
     "torchaudio>=0.8.1",
-    "einops>=0.7.0",
+    "einops>=0.6.1",
     "transformers_stream_generator>=0.0.4",
-    "peft>=0.7.0",
+    "peft>=0.3.0",
     "datasets>=2.15.0",
-    "huggingface-hub>=0.19.4",
+    "huggingface-hub>=0.16.4",
 ]
 
 setuptools.setup(