From 4905109240a79801f643952572460486fd9dc54a Mon Sep 17 00:00:00 2001 From: Jon Perry Date: Fri, 26 Jul 2024 11:13:45 -0400 Subject: [PATCH] chore: install dev deps for downloading models during e2e workflows --- .github/workflows/e2e-llama-cpp-python.yaml | 4 +++- .github/workflows/e2e-text-embeddings.yaml | 4 +++- .github/workflows/e2e-vllm.yaml | 4 +++- .github/workflows/e2e-whisper.yaml | 4 +++- 4 files changed, 12 insertions(+), 4 deletions(-) diff --git a/.github/workflows/e2e-llama-cpp-python.yaml b/.github/workflows/e2e-llama-cpp-python.yaml index 12f1e1fac8..9a640cf684 100644 --- a/.github/workflows/e2e-llama-cpp-python.yaml +++ b/.github/workflows/e2e-llama-cpp-python.yaml @@ -64,7 +64,9 @@ jobs: python-version-file: 'pyproject.toml' - name: Install Python Deps - run: python -m pip install "." + run: | + python -m pip install "." + python -m pip install -U "huggingface_hub[cli,hf_transfer]" - name: Setup UDS Environment uses: defenseunicorns/uds-common/.github/actions/setup@05f42bb3117b66ebef8c72ae050b34bce19385f5 diff --git a/.github/workflows/e2e-text-embeddings.yaml b/.github/workflows/e2e-text-embeddings.yaml index f11acd5ef7..c5034bdf35 100644 --- a/.github/workflows/e2e-text-embeddings.yaml +++ b/.github/workflows/e2e-text-embeddings.yaml @@ -66,7 +66,9 @@ jobs: python-version-file: 'pyproject.toml' - name: Install Python Deps - run: python -m pip install "." + run: | + python -m pip install "." + python -m pip install -U "huggingface_hub[cli,hf_transfer]" - name: Setup UDS Environment uses: defenseunicorns/uds-common/.github/actions/setup@05f42bb3117b66ebef8c72ae050b34bce19385f5 diff --git a/.github/workflows/e2e-vllm.yaml b/.github/workflows/e2e-vllm.yaml index caa52d1deb..3f95965f22 100644 --- a/.github/workflows/e2e-vllm.yaml +++ b/.github/workflows/e2e-vllm.yaml @@ -66,7 +66,9 @@ jobs: python-version-file: 'pyproject.toml' - name: Install Python Deps - run: python -m pip install "." + run: | + python -m pip install "." + python -m pip install -U "huggingface_hub[cli,hf_transfer]" - name: Setup UDS Environment uses: defenseunicorns/uds-common/.github/actions/setup@05f42bb3117b66ebef8c72ae050b34bce19385f5 diff --git a/.github/workflows/e2e-whisper.yaml b/.github/workflows/e2e-whisper.yaml index d93f215fc5..c82f513e77 100644 --- a/.github/workflows/e2e-whisper.yaml +++ b/.github/workflows/e2e-whisper.yaml @@ -64,7 +64,9 @@ jobs: python-version-file: 'pyproject.toml' - name: Install Python Deps - run: python -m pip install "." + run: | + python -m pip install "." + python -m pip install -U ctranslate2 "transformers[torch]" - name: Setup UDS Environment uses: defenseunicorns/uds-common/.github/actions/setup@05f42bb3117b66ebef8c72ae050b34bce19385f5