From 2c64dc967c66d5b482be2e9857698267e585a02a Mon Sep 17 00:00:00 2001 From: Niels Bantilan Date: Tue, 11 Jun 2024 09:04:23 -0700 Subject: [PATCH] updates Signed-off-by: Niels Bantilan --- .github/workflows/fine_tune.yaml | 8 +++----- llm_edge_finetuning/tasks_training.py | 2 +- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/.github/workflows/fine_tune.yaml b/.github/workflows/fine_tune.yaml index af537f7..ef00284 100644 --- 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 index c703806..bf4367b 100644 --- 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"),