Fix Inf2 package dependency version conflicts (#2887)

* fix torchdata version * Skip specifying versions for protobuf and grpcio-tools * skip specifying grpcio-tools * test inf2 dependency fix for benchmark * Revert "test inf2 dependency fix for benchmark" This reverts commit 8899310.
pytorch · Jan 3, 2024 · 2eae8e5 · 2eae8e5
1 parent 030ee8d
commit 2eae8e5
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/requirements/neuronx.txt b/requirements/neuronx.txt
@@ -1,7 +1,4 @@
 --extra-index-url https://pip.repos.neuron.amazonaws.com
-numpy==1.22.4
-protobuf==4.25.1
-grpcio-tools==1.48.2
 neuronx-cc
 torch-neuronx
 transformers-neuronx
diff --git a/requirements/torch_neuronx_linux.txt b/requirements/torch_neuronx_linux.txt
@@ -4,4 +4,4 @@ torch==1.13.1+cpu
 torchvision==0.14.1+cpu
 torchtext==0.14.1
 torchaudio==0.13.1+cpu
-torchdata==0.7.0
+torchdata==0.5.1