Skip to content

Commit eaacf9d

Browse files
authored
pin the gpt-fast repo commit version (#2966)
1 parent a0a17a9 commit eaacf9d

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

examples/large_models/gpt_fast/README.md

+6
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,17 @@ The examples has been tested on A10, A100 as well as H100.
2323
Install dependencies and upgrade torch to nightly build (currently required)
2424
```
2525
git clone https://github.com/pytorch-labs/gpt-fast/
26+
git checkout f44ef4eb55b54ec4c452b669eee409421adabd60
2627
pip install sentencepiece huggingface_hub
2728
pip uninstall torchtext torchdata torch torchvision torchaudio -y
2829
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121 --ignore-installed
2930
```
3031

32+
You can also install PyTorch nightlies using the below command
33+
```
34+
python ./ts_scripts/install_dependencies.py --cuda=cu121 --nightly_torch
35+
```
36+
3137
### Step 1: Download and convert the weights
3238

3339
Currently supported models:

0 commit comments

Comments
 (0)