Fix optimum-tpu pip install instructions (#29)

huggingface · Apr 26, 2024 · 5e5a1d7 · 5e5a1d7
1 parent c7fe483
commit 5e5a1d7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -31,7 +31,7 @@ We currently support a few LLM models targeting text generation scenarios:
 
 `optimum-tpu` comes with an handy PyPi released package compatible with your classical python dependency management tool.
 
-`pip install optimum-tpu`
+`pip install optimum-tpu -f https://storage.googleapis.com/libtpu-releases/index.html`
 
 ## Inference
 

diff --git a/docs/source/howto/deploy.mdx b/docs/source/howto/deploy.mdx
@@ -65,7 +65,7 @@ $ >
 If you want to leverage PyTorch/XLA through Optimum-TPU, it should be as simple as
 
 ```bash
-$ python3 -m pip install optimum-tpu
+$ python3 -m pip install optimum-tpu -f https://storage.googleapis.com/libtpu-releases/index.html
 $ export PJRT_DEVICE=TPU
 ```