ping neuron repos

huggingface · Oct 27, 2023 · 5e6659d · 5e6659d
1 parent 8adbf01
commit 5e6659d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -45,20 +45,20 @@
     "quality": QUALITY_REQUIRES,
     "neuron": [
         "wheel",
-        "torch-neuron==1.13.1.*",
+        "torch-neuron==1.13.1.2.9.1.0",
         "torch==1.13.1.*",
-        "neuron-cc[tensorflow]==1.18.*",
+        "neuron-cc[tensorflow]==1.19.0.0",
         "protobuf",
         "torchvision",
     ],
     "neuronx": [
         "wheel",
-        "neuronx-cc>=2.10",
-        "torch-neuronx>=1.13.1.1.11.0",
-        "transformers-neuronx>=0.7.84",
+        "neuronx-cc==2.10.0.35",
+        "torch-neuronx==1.13.1.1.11.0",
+        "transformers-neuronx==0.7.84",
         "torch==1.13.1.*",
         "torchvision==0.14.*",
-        "neuronx_distributed>=0.4.0",
+        "neuronx_distributed==0.4.0",
     ],
     "diffusers": ["diffusers"],
 }