updates

Signed-off-by: Niels Bantilan <[email protected]>
unionai-oss · Jun 11, 2024 · 2c64dc9 · 2c64dc9
1 parent bb595e2
commit 2c64dc9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/.github/workflows/fine_tune.yaml b/.github/workflows/fine_tune.yaml
@@ -19,14 +19,12 @@ jobs:
       - name: Set up Python 3.11
         uses: actions/setup-python@v5
       - name: Install dependencies
-        run:
-          pip install uv
-          pip install -U setuptools
-          uv pip install -r requirements.lock.txt
+        run: pip install -r requirements.lock.txt
       - name: Run fine-tuning
+        env:
+          UNIONAI_SERVERLESS_API_KEY: ${{ secrets.UNIONAI_SERVERLESS_API_KEY_GCP }}
         run:
           unionai run --copy-all --remote \
             llm_edge_finetuning/workflows_news.py train_workflow \
             --config config/phi_3_mini_128k_instruct.json \
             --categories '["science", "technology", "business"]'
-
diff --git a/llm_edge_finetuning/tasks_training.py b/llm_edge_finetuning/tasks_training.py
@@ -23,7 +23,7 @@
 @task(
     retries=3,
     cache=True,
-    cache_version="12",
+    cache_version="13",
     container_image=image_spec,
     accelerator=accelerators.T4,
     requests=Resources(mem="24Gi", cpu="4", gpu="1"),