8000 pin the gpt-fast repo commit version by agunapal · Pull Request #2966 · pytorch/serve · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
8000

pin the gpt-fast repo commit version #2966

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Feb 27, 2024
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions examples/large_models/gpt_fast/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,11 +23,17 @@ The examples has been tested on A10, A100 as well as H100.
Install dependencies and upgrade torch to nightly build (currently required)
```
git clone https://github.com/pytorch-labs/gpt-fast/
git checkout f44ef4eb55b54ec4c452b669eee409421adabd60
pip install sentencepiece huggingface_hub
pip uninstall torchtext torchdata torch torchvision torchaudio -y
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121 --ignore-installed
```

You can also install PyTorch nightlies using the below command
```
python ./ts_scripts/install_dependencies.py --cuda=cu121 --nightly_torch
```

### Step 1: Download and convert the weights

Currently supported models:
Expand Down
0