fix integ tests

dlc_developer_config.toml

@@ -9,7 +9,7 @@ ei_mode = false
 neuron_mode = false
 # Please only set it to true if you are preparing a NEURONX related PR
 # Do remember to revert it back to false before merging any PR (including NEURONX dedicated PR)
-neuronx_mode = true
+neuronx_mode = false
 # Please only set it to true if you are preparing a GRAVITON related PR
 # Do remember to revert it back to false before merging any PR (including GRAVITON dedicated PR)
 graviton_mode = false
@@ -34,10 +34,10 @@ deep_canary_mode = false
 [build]
 # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
 # available frameworks - ["autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "mxnet", "pytorch", "stabilityai_pytorch"]
-build_frameworks = ["huggingface_pytorch"]
+build_frameworks = []
 
 # By default we build both training and inference containers. Set true/false values to determine which to build.
-build_training = false
+build_training = true
 build_inference = true
 
 # Set do_build to "false" to skip builds and test the latest image built by this PR
@@ -143,7 +143,7 @@ dlc-pr-tensorflow-2-neuron-inference = ""
 # HuggingFace Inference
 dlc-pr-huggingface-tensorflow-inference = ""
 dlc-pr-huggingface-pytorch-inference = ""
-dlc-pr-huggingface-pytorch-neuron-inference = "huggingface/pytorch/inference/buildspec-1-13-1-neuronx.yml"
+dlc-pr-huggingface-pytorch-neuron-inference = ""
 
 # Stability AI Inference
 dlc-pr-stabilityai-pytorch-inference = ""

huggingface/pytorch/inference/docker/1.13/py3/sdk2.16.1/Dockerfile.neuronx

@@ -24,6 +24,7 @@ ARG TRANSFORMERS_VERSION
 ARG DIFFUSERS_VERSION=0.25.1
 ARG OPTIMUM_NEURON_VERSION=0.0.18
 ARG SENTENCE_TRANSFORMERS=2.2.2
+ARG HUGGINGFACE_HUB_VERSION=0.25.2
 
 # See http://bugs.python.org/issue19846
 ENV LANG C.UTF-8
@@ -176,6 +177,7 @@ RUN pip install --no-cache-dir \
 	transformers[sentencepiece,audio,vision]==${TRANSFORMERS_VERSION} \
     diffusers==${DIFFUSERS_VERSION} \
     optimum-neuron==${OPTIMUM_NEURON_VERSION} \
+    huggingface_hub==${HUGGINGFACE_HUB_VERSION} \
     sentence_transformers==${SENTENCE_TRANSFORMERS} \
 	"sagemaker-huggingface-inference-toolkit<3"