File tree 1 file changed +6
-0
lines changed
examples/large_models/gpt_fast
1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -23,11 +23,17 @@ The examples has been tested on A10, A100 as well as H100.
23
23
Install dependencies and upgrade torch to nightly build (currently required)
24
24
```
25
25
git clone https://github.com/pytorch-labs/gpt-fast/
26
+ git checkout f44ef4eb55b54ec4c452b669eee409421adabd60
26
27
pip install sentencepiece huggingface_hub
27
28
pip uninstall torchtext torchdata torch torchvision torchaudio -y
28
29
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121 --ignore-installed
29
30
```
30
31
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
+
31
37
### Step 1: Download and convert the weights
32
38
33
39
Currently supported models:
You can’t perform that action at this time.
0 commit comments