Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -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"
]
Expand Down
2 changes: 1 addition & 1 deletion torch-neuron/inference/fairseq/Fairseq.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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"
]
},
{
Expand Down
2 changes: 1 addition & 1 deletion torch-neuron/inference/ssd/SSD300VGG16.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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"
]
},
{
Expand Down
2 changes: 1 addition & 1 deletion torch-neuronx/training/hf_image_classification/vit.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
Loading