From e5ac498b8dbce1b254032ab9cc066cfb0a0d303d Mon Sep 17 00:00:00 2001 From: Andreea Popescu Date: Fri, 6 Sep 2024 10:45:30 +0800 Subject: [PATCH] wip --- .github/workflows/build_push_image.yml | 2 +- .github/workflows/smoke_test.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build_push_image.yml b/.github/workflows/build_push_image.yml index 8fe3a8c..74be7cc 100644 --- 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 index da9979a..fa1062a 100644 --- 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/