Improve installation guide (#559)

improve installation guide
huggingface · Apr 9, 2024 · 57fc9a5 · 57fc9a5
1 parent bb1cc96
commit 57fc9a5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -26,13 +26,13 @@ To install the latest release of this package:
 * For AWS Trainium (trn1) or AWS inferentia2 (inf2)
 
 ```bash
-pip install optimum[neuronx]
+pip install --upgrade-strategy eager optimum[neuronx]
 ```
 
 * For AWS inferentia (inf1)
 
 ```bash
-pip install optimum[neuron]
+pip install --upgrade-strategy eager optimum[neuron]
 ```
 
 Optimum Neuron is a fast-moving project, and you may want to install it from source:

diff --git a/docs/source/installation.mdx b/docs/source/installation.mdx
@@ -29,11 +29,11 @@ python -m pip config set global.extra-index-url https://pip.repos.neuron.amazona
 ## Installing `optimum-neuron` for AWS Trainium (`trn1`) or AWS inferentia2 (`inf2`)
 
 ```bash
-python -m pip install optimum[neuronx]
+python -m pip install --upgrade-strategy eager optimum[neuronx]
 ```
 
 ## Installing `optimum-neuron` for AWS inferentia (`inf1`)
 
 ```bash
-python -m pip install optimum[neuron]
+python -m pip install --upgrade-strategy eager optimum[neuron]
 ```