From a2aa568927ce3c9ac537e0c286706286c50663ff Mon Sep 17 00:00:00 2001 From: Andreea Popescu Date: Mon, 16 Sep 2024 14:32:42 +0800 Subject: [PATCH] push image --- .github/workflows/build_push_image.yml | 6 ++++-- src/compute_horde_prompt_gen/model.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/.github/workflows/build_push_image.yml b/.github/workflows/build_push_image.yml index c38630c..3bf1d11 100644 --- a/.github/workflows/build_push_image.yml +++ b/.github/workflows/build_push_image.yml @@ -2,7 +2,7 @@ name: "CD: build & push image" on: push: - branches: [build-image] + branches: [build-push-llama-image] workflow_dispatch: env: @@ -38,6 +38,8 @@ jobs: cd src/compute_horde_prompt_gen - python download_model.py --model_name phi3 --huggingface_token "${{ secrets.HUGGINGFACE_API_KEY }}" + python download_model.py --model_name llama3 --huggingface_token "${{ secrets.HUGGINGFACE_API_KEY }}" docker build -t $IMAGE_NAME . + + docker push $IMAGE_NAME:TAG_VERSION diff --git a/src/compute_horde_prompt_gen/model.py b/src/compute_horde_prompt_gen/model.py index d68b513..abac394 100644 --- a/src/compute_horde_prompt_gen/model.py +++ b/src/compute_horde_prompt_gen/model.py @@ -15,7 +15,7 @@ def __init__(self): pass def generate(self, prompts: list[str], num_return_sequences: int, **_kwargs): - content = f"Here is the list of prompts:\nHow are you?\nDescribe something\nCount to ten\n" + content = "Here is the list of prompts:\nHow are you?\nDescribe something\nCount to ten\n" return [content for _ in range(len(prompts) * num_return_sequences)]