Retrigger workflow

pytorch · Dec 20, 2023 · bfea7b3 · bfea7b3
1 parent 9d3e531
commit bfea7b3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/temp_workflow.yml b/.github/workflows/temp_workflow.yml
@@ -46,6 +46,8 @@ jobs:
           cd /home/ubuntu
           git clone https://github.com/pytorch-labs/gpt-fast/
           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
           export MODEL_REPO=meta-llama/Llama-2-7b-chat-hf
           cd gpt-fast
           huggingface-cli login --token $HUGGING_FACE_HUB_TOKEN