From ec8f155bcd26ccc39bb50f09808681085ea1497f Mon Sep 17 00:00:00 2001
From: Andreea Popescu <andreea.popescu@reef.pl>
Date: Wed, 4 Sep 2024 17:23:02 +0800
Subject: [PATCH] f

---
 .github/workflows/build_push_image.yml | 6 +++---
 .github/workflows/smoke_test.yml       | 9 +++++----
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/build_push_image.yml b/.github/workflows/build_push_image.yml
index 096999a..1630f21 100644
--- a/.github/workflows/build_push_image.yml
+++ b/.github/workflows/build_push_image.yml
@@ -41,7 +41,7 @@ jobs:
           echo "-----"
           ls
           echo "-----"
-          pdm install
+          pdm install transformers
           echo "----- done"
 
       - name: Docker build and push
@@ -60,6 +60,6 @@ jobs:
           ls
           echo "-----"
 
-          docker build -t "${{ secrets.HUGGINGFACE_API_KEY }}" -horde-prompt-gen .
+          docker build -t "${{ IMAGE_NAME }}" -horde-prompt-gen .
 
-          docker run -v ./output/:/app/output/ "${{ secrets.HUGGINGFACE_API_KEY }}" --number_of_batches 5 --number_of_prompts_per_batch 30 --uuids uuid1,uuid2,uuid3,uuid4,uuid5
+          docker run -v ./output/:/app/output/ "${{ IMAGE_NAME }}" --number_of_batches 5 --number_of_prompts_per_batch 30 --uuids uuid1,uuid2,uuid3,uuid4,uuid5
diff --git a/.github/workflows/smoke_test.yml b/.github/workflows/smoke_test.yml
index a4f447e..42d69b4 100644
--- a/.github/workflows/smoke_test.yml
+++ b/.github/workflows/smoke_test.yml
@@ -23,14 +23,15 @@ jobs:
         with:
           python-version: ${{ env.PYTHON_DEFAULT_VERSION }}
 
-      - name: Install dependencies
-        run: | 
-          python -m pip install --upgrade 'pdm>=2.12,<3'
-          pdm install
+      # - name: Install dependencies
+        # run: | 
+          # python -m pip install --upgrade 'pdm>=2.12,<3'
+          # pdm install
 
       - name: Run Test
         run: | 
           cd src/compute_horde_prompt_gen
+          mkdir saved_models/
           mkdir output/
 
           echo "building image with no model"