f

backend-developers-ltd · Sep 4, 2024 · 0590fae · 0590fae
1 parent fde42c3
commit 0590fae
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/.github/workflows/build_push_image.yml b/.github/workflows/build_push_image.yml
@@ -36,15 +36,16 @@ jobs:
 
       - name: Install dependencies
         run: | 
-          python -m pip install --upgrade 'pdm>=2.12,<3'
-          pdm install transformers
+          # python -m pip install --upgrade 'pdm>=2.12,<3'
+          # pdm install transformers
+          python -m pip install transformers torch
 
       - name: Docker build and push
         run: |
           df -h
           IMAGE_NAME="${DOCKER_REPO_NAME}:${TAG_VERSION}"
 
-          pdm run download_model.py --huggingface_token "${{ secrets.HUGGINGFACE_API_KEY }}"
+          python download_model.py --huggingface_token "${{ secrets.HUGGINGFACE_API_KEY }}"
 
           cd src/compute_horde_prompt_gen
           ls

diff --git a/.github/workflows/smoke_test.yml b/.github/workflows/smoke_test.yml
@@ -33,7 +33,7 @@ jobs:
         run: | 
           cd src/compute_horde_prompt_gen
 
-          pdm run run.py --mock_model --number_of_batches 5 --number_of_prompts_per_batch 20 --uuids uuid1,uuid2,uuid3,uuid4,uuid5
+          python3 run.py --mock_model --number_of_batches 5 --number_of_prompts_per_batch 20 --uuids uuid1,uuid2,uuid3,uuid4,uuid5
 
           # mkdir saved_models/
           # mkdir output/