README.md: use PyPI for installation instead of JFrog
Browse files
README.md
CHANGED
|
@@ -126,17 +126,11 @@ To work with our models just run these lines in your terminal:
|
|
| 126 |
|
| 127 |
```shell
|
| 128 |
pip install thestage
|
| 129 |
-
pip install
|
| 130 |
-
--index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple\
|
| 131 |
-
--extra-index-url https://pypi.nvidia.com\
|
| 132 |
-
--extra-index-url https://pypi.org/simple
|
| 133 |
pip install flash_attn==2.7.3 --no-build-isolation
|
| 134 |
|
| 135 |
# or for blackwell support
|
| 136 |
-
pip install
|
| 137 |
-
--index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple\
|
| 138 |
-
--extra-index-url https://pypi.nvidia.com\
|
| 139 |
-
--extra-index-url https://pypi.org/simple
|
| 140 |
pip install torch==2.7.0+cu128 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
|
| 141 |
# please download the appropriate version of Wheels for your system from https://github.com/Zarrac/flashattention-blackwell-wheels-whl-ONLY-5090-5080-5070-5060-flash-attention-/releases/tag/FlashAttention
|
| 142 |
mv flash_attn-2.7.4.post1-rtx5090-torch2.7.0cu128cxx11abiTRUE-cp311-linux_x86_64.whl flash_attn-2.7.4.post1-0rtx5090torch270cu128cxx11abiTRUE-cp311-cp311-linux_x86_64.whl
|
|
|
|
| 126 |
|
| 127 |
```shell
|
| 128 |
pip install thestage
|
| 129 |
+
pip install 'thestage-elastic-models[nvidia]'
|
|
|
|
|
|
|
|
|
|
| 130 |
pip install flash_attn==2.7.3 --no-build-isolation
|
| 131 |
|
| 132 |
# or for blackwell support
|
| 133 |
+
pip install 'thestage-elastic-models[blackwell]'
|
|
|
|
|
|
|
|
|
|
| 134 |
pip install torch==2.7.0+cu128 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
|
| 135 |
# please download the appropriate version of Wheels for your system from https://github.com/Zarrac/flashattention-blackwell-wheels-whl-ONLY-5090-5080-5070-5060-flash-attention-/releases/tag/FlashAttention
|
| 136 |
mv flash_attn-2.7.4.post1-rtx5090-torch2.7.0cu128cxx11abiTRUE-cp311-linux_x86_64.whl flash_attn-2.7.4.post1-0rtx5090torch270cu128cxx11abiTRUE-cp311-cp311-linux_x86_64.whl
|