From e8f1bba84ce4a53f23abbcbaae2b787c2fb9ca45 Mon Sep 17 00:00:00 2001 From: "(AWS) Rahul Huilgol" Date: Tue, 2 Dec 2025 09:25:20 -0800 Subject: [PATCH 1/4] Remove unnecessary pins --- torch-neuronx/training/hf_image_classification/vit.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch-neuronx/training/hf_image_classification/vit.ipynb b/torch-neuronx/training/hf_image_classification/vit.ipynb index 4ba19001..f97eff34 100644 --- a/torch-neuronx/training/hf_image_classification/vit.ipynb +++ b/torch-neuronx/training/hf_image_classification/vit.ipynb @@ -40,7 +40,7 @@ "source": [ "%env TOKENIZERS_PARALLELISM=True #Supresses tokenizer warnings making errors easier to detect\n", "#Install Neuron Compiler and Neuron/XLA packages\n", - "%pip install -U \"protobuf<4\" \"transformers==4.52.3\" \"numpy==1.24.4\" optimum-neuron datasets evaluate scikit-learn \n", + "%pip install -U protobuf \"transformers==4.52.3\" numpy optimum-neuron datasets evaluate scikit-learn \n", "%pip install -U \"accelerate==1.7.0\"\n", "# use --force-reinstall if you're facing some issues while loading the modules\n", "# now restart the kernel again" From a8867ad150223964e81a473b1b682d0470e5ab86 Mon Sep 17 00:00:00 2001 From: "(AWS) Rahul Huilgol" Date: Tue, 2 Dec 2025 10:53:03 -0800 Subject: [PATCH 2/4] Update VisionPerceiverConv.ipynb --- archive/hf_image_classification/VisionPerceiverConv.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/archive/hf_image_classification/VisionPerceiverConv.ipynb b/archive/hf_image_classification/VisionPerceiverConv.ipynb index 92d035fd..11c358d9 100644 --- a/archive/hf_image_classification/VisionPerceiverConv.ipynb +++ b/archive/hf_image_classification/VisionPerceiverConv.ipynb @@ -40,7 +40,7 @@ "source": [ "%env TOKENIZERS_PARALLELISM=True #Supresses tokenizer warnings making errors easier to detect\n", "#Install Neuron Compiler and Neuron/XLA packages\n", - "%pip install -U \"protobuf<4\" \"transformers==4.52.3\" \"accelerate==1.7.0\" \"numpy==1.24.4\" optimum-neuron evaluate datasets scikit-learn \n", + "%pip install -U protobuf \"transformers==4.52.3\" \"accelerate==1.7.0\" numpy optimum-neuron evaluate datasets scikit-learn \n", "# use --force-reinstall if you're facing some issues while loading the modules\n", "# now restart the kernel again" ] From a9e27cf0c84dcbbf37150e4743fb3276059cbbbb Mon Sep 17 00:00:00 2001 From: "(AWS) Rahul Huilgol" Date: Tue, 2 Dec 2025 10:53:37 -0800 Subject: [PATCH 3/4] Remove version constraint for protobuf in notebook --- torch-neuron/inference/ssd/SSD300VGG16.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch-neuron/inference/ssd/SSD300VGG16.ipynb b/torch-neuron/inference/ssd/SSD300VGG16.ipynb index 4df7c194..74aa7abe 100644 --- a/torch-neuron/inference/ssd/SSD300VGG16.ipynb +++ b/torch-neuron/inference/ssd/SSD300VGG16.ipynb @@ -35,7 +35,7 @@ }, "outputs": [], "source": [ - "%pip install \"protobuf<4\" numpy==1.20.0 torch-neuron tensorflow==1.15 neuron-cc" + "%pip install protobuf numpy torch-neuron tensorflow==1.15 neuron-cc" ] }, { From 75ea9bf1e5a2e19ff08f85977881d9fb5feae346 Mon Sep 17 00:00:00 2001 From: "(AWS) Rahul Huilgol" Date: Tue, 2 Dec 2025 10:54:06 -0800 Subject: [PATCH 4/4] Fix pip install command by removing numpy version --- torch-neuron/inference/fairseq/Fairseq.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch-neuron/inference/fairseq/Fairseq.ipynb b/torch-neuron/inference/fairseq/Fairseq.ipynb index b907dde5..8b7a27d3 100644 --- a/torch-neuron/inference/fairseq/Fairseq.ipynb +++ b/torch-neuron/inference/fairseq/Fairseq.ipynb @@ -63,7 +63,7 @@ " !git clone https://github.com/pytorch/fairseq fairseq-local && \\\n", " cd fairseq-local && git checkout acd9a53607d1e5c64604e88fc9601d0ee56fd6f1 && \\\n", " pip3 install --editable ./ && \\\n", - " pip3 --no-cache-dir install sacremoses torch==1.11.0 torchaudio==0.11.0 \"numpy==1.22.2\" scikit-learn fastBPE" + " pip3 --no-cache-dir install sacremoses torch==1.11.0 torchaudio==0.11.0 numpy scikit-learn fastBPE" ] }, {