From 57fc9a5112c5c30ccbf8a75efac87823908aef42 Mon Sep 17 00:00:00 2001
From: Jingya HUANG <44135271+JingyaHuang@users.noreply.github.com>
Date: Tue, 9 Apr 2024 13:47:52 +0200
Subject: [PATCH] Improve installation guide (#559)

improve installation guide
---
 README.md                    | 4 ++--
 docs/source/installation.mdx | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index b8d3e9b0f..dbc9b0a26 100644
--- 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
index cb298051a..1758f1a72 100644
--- 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]
 ```