From 6fc398613702f6dba016964280b4b139d09d4933 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Tue, 3 Sep 2024 09:43:27 -0700 Subject: [PATCH] Undo pinning transformers --- .github/workflows/nv-a6000-fastgen.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/nv-a6000-fastgen.yml b/.github/workflows/nv-a6000-fastgen.yml index a820794d..80bca5ae 100644 --- a/.github/workflows/nv-a6000-fastgen.yml +++ b/.github/workflows/nv-a6000-fastgen.yml @@ -35,9 +35,8 @@ jobs: python -c "import torch; print('CUDA available:', torch.cuda.is_available())" - name: Install transformers run: | - git clone https://github.com/huggingface/transformers + git clone --depth=1 https://github.com/huggingface/transformers cd transformers - git checkout v4.44.2 git rev-parse --short HEAD python -m pip install . - name: Install deepspeed