Update README.md (#4)
Browse files- Update README.md (2b4b29b64809c72a4faf5e5a7328bc1bf2fcce95)
Co-authored-by: Alex <bratpidd@users.noreply.huggingface.co>
README.md
CHANGED
|
@@ -112,7 +112,7 @@ To work with our elastic models and compilation tools, you\'ll need to install `
|
|
| 112 |
|
| 113 |
```shell
|
| 114 |
pip install thestage
|
| 115 |
-
pip install 'thestage-elastic-models[nvidia]'
|
| 116 |
pip install flash-attn==2.7.3 --no-build-isolation
|
| 117 |
pip uninstall apex
|
| 118 |
```
|
|
|
|
| 112 |
|
| 113 |
```shell
|
| 114 |
pip install thestage
|
| 115 |
+
pip install 'thestage-elastic-models[nvidia]' --extra-index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple
|
| 116 |
pip install flash-attn==2.7.3 --no-build-isolation
|
| 117 |
pip uninstall apex
|
| 118 |
```
|