hinairo commited on
Commit
160019c
·
verified ·
1 Parent(s): fedd277

README.md: use PyPI for installation instead of JFrog

Browse files
Files changed (1) hide show
  1. README.md +2 -8
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 elastic_models[nvidia]\
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 elastic_models[blackwell]\
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