From 9f313106e651797df3802bff7ed79d6ee5694aa5 Mon Sep 17 00:00:00 2001 From: agunapal Date: Mon, 26 Feb 2024 23:08:12 +0000 Subject: [PATCH] pin the gpt-fast repo commit version --- examples/large_models/gpt_fast/README.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/examples/large_models/gpt_fast/README.md b/examples/large_models/gpt_fast/README.md index d495d4813d..a1ec8bb29d 100644 --- a/examples/large_models/gpt_fast/README.md +++ b/examples/large_models/gpt_fast/README.md @@ -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: