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" ] 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" ] }, { 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" ] }, { 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"