From db27613072501c6f60652043637bd9973757f186 Mon Sep 17 00:00:00 2001 From: Tyler Osterberg Date: Tue, 12 Dec 2023 11:03:34 -0800 Subject: [PATCH] [fix] inf2 container freeze compiler versions --- serving/docker/pytorch-inf2.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile index 9a281b5f0..b7ed4c4a5 100644 --- a/serving/docker/pytorch-inf2.Dockerfile +++ b/serving/docker/pytorch-inf2.Dockerfile @@ -17,7 +17,7 @@ ARG neuronsdk_version=2.14.1 ARG torch_neuronx_version=1.13.1.1.12.0 ARG transformers_neuronx_version=0.8.268 ARG neuronx_distributed_version=0.5.0 -ARG neuronx_cc_version=2.11.* +ARG neuronx_cc_version=2.11.0.35 ARG protobuf_version=3.20.3 ARG transformers_version=4.34.0 ARG accelerate_version=0.23.0