f

backend-developers-ltd · Sep 4, 2024 · ec8f155 · ec8f155
1 parent 9cceac0
commit ec8f155
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git 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
@@ -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"