push image

backend-developers-ltd · Sep 16, 2024 · e502cd6 · e502cd6
1 parent ddcc0a9
commit e502cd6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git 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:
@@ -14,7 +14,7 @@ jobs:
   deploy:
     timeout-minutes: 15
     runs-on:
-      group: bulkier
+      group: bulkierer
     steps:
       - uses: actions/checkout@v2
         with:
@@ -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
diff --git 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)]