chore: bump neuronx-cc version

huggingface · Jan 30, 2024 · e5c7dee · e5c7dee
1 parent 41f73ea
commit e5c7dee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -55,7 +55,7 @@
     ],
     "neuronx": [
         "wheel",
-        "neuronx-cc==2.12.54.0",
+        "neuronx-cc==2.12.68.0",
         "torch-neuronx==1.13.1.1.13.0",
         "transformers-neuronx==0.9.474",
         "torch==1.13.1.*",

diff --git a/text-generation-inference/Dockerfile b/text-generation-inference/Dockerfile
@@ -99,7 +99,7 @@ RUN apt-get update -y \
 ENV PATH="/opt/bin/:/opt/aws/neuron/bin:${PATH}"
 
 RUN pip3 install \
-    neuronx-cc==2.12.54.0 \
+    neuronx-cc==2.12.68.0 \
     torch-neuronx==1.13.1.1.13.0 \
     transformers-neuronx==0.9.474 \
     --extra-index-url=https://pip.repos.neuron.amazonaws.com