From e26b5c34c0df02067923b4b85bb0a4340847ce5b Mon Sep 17 00:00:00 2001
From: Michael Benayoun <mickbenayoun@gmail.com>
Date: Fri, 4 Aug 2023 12:38:23 +0200
Subject: [PATCH] Update setup.py

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 3f44a7533..a84a2bd43 100644
--- 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.*",