Update setup.py

huggingface · Aug 4, 2023 · e26b5c3 · e26b5c3
1 parent 1dafbf1
commit e26b5c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -53,7 +53,7 @@
     ],
     "neuronx": [
         "wheel",
-        "neuronx-cc==2.6.*",
+        "neuronx-cc==2.*",
         "torch-neuronx",
         "transformers-neuronx",
         "torch==1.13.1.*",