Ilba bratpidd commited on
Commit
9761676
·
verified ·
1 Parent(s): 2e610fa

Update README.md (#6)

Browse files

- Update README.md (e22ce806390e6b31540371f6633a6ad57656d42d)


Co-authored-by: Alex <bratpidd@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -121,11 +121,11 @@ To work with our models just run these lines in your terminal:
121
 
122
  ```shell
123
  pip install thestage
124
- pip install 'thestage-elastic-models[nvidia]'
125
  pip install flash_attn==2.7.3 --no-build-isolation
126
 
127
  # or for blackwell support
128
- pip install 'thestage-elastic-models[blackwell]'
129
  pip install torch==2.7.0+cu128 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
130
  # 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
131
  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
 
121
 
122
  ```shell
123
  pip install thestage
124
+ pip install 'thestage-elastic-models[nvidia]' --extra-index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple
125
  pip install flash_attn==2.7.3 --no-build-isolation
126
 
127
  # or for blackwell support
128
+ pip install 'thestage-elastic-models[blackwell]' --extra-index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple
129
  pip install torch==2.7.0+cu128 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
130
  # 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
131
  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