wip

backend-developers-ltd · Sep 6, 2024 · e5ac498 · e5ac498
1 parent 6029589
commit e5ac498
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build_push_image.yml b/.github/workflows/build_push_image.yml
@@ -44,7 +44,7 @@ jobs:
           IMAGE_NAME="${DOCKER_REPO_NAME}:${TAG_VERSION}"
 
           cd src/compute_horde_prompt_gen
-          python download_model.py --huggingface_token "${{ secrets.HUGGINGFACE_API_KEY }}"
+          python download_model.py --model_name phi3 --huggingface_token "${{ secrets.HUGGINGFACE_API_KEY }}"
 
           ls
           docker build -t $IMAGE_NAME .

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