chore: use AWS Neuron SDK 2.21.0 pip packages

huggingface · Dec 23, 2024 · 61981cf · 61981cf
1 parent a03ed5b
commit 61981cf
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/setup.py b/setup.py
@@ -64,13 +64,13 @@
     ],
     "neuronx": [
         "wheel",
-        "neuronx-cc==2.15.143.0",
-        "torch-neuronx==2.1.2.2.3.2",
-        "transformers-neuronx==0.12.313",
-        "torch==2.1.2.*",
-        "torchvision==0.16.*",
-        "neuronx_distributed==0.9.0",
-        "libneuronxla==2.0.5347.0",
+        "neuronx-cc==2.16.345.0",
+        "torch-neuronx==2.5.1.2.4.0",
+        "transformers-neuronx==0.13.322",
+        "torch==2.5.1.*",
+        "torchvision==0.20.*",
+        "neuronx_distributed==0.10.0",
+        "libneuronxla==2.1.681.0",
     ],
     "diffusers": ["diffusers>=0.28.0, <=0.30.3", "peft"],
     "sentence-transformers": ["sentence-transformers >= 2.2.0"],