From 0590fae60e35617f866d43bb5c87a92ff462c766 Mon Sep 17 00:00:00 2001
From: Andreea Popescu <andreea.popescu@reef.pl>
Date: Wed, 4 Sep 2024 18:36:08 +0800
Subject: [PATCH] f

---
 .github/workflows/build_push_image.yml | 7 ++++---
 .github/workflows/smoke_test.yml       | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build_push_image.yml b/.github/workflows/build_push_image.yml
index 512f0c0..94c7838 100644
--- 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
index 8428fd5..64a8d8a 100644
--- 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/