diff --git a/.github/workflows/fine_tune.yaml b/.github/workflows/fine_tune.yaml index fb13a8e..13949ce 100644 --- a/.github/workflows/fine_tune.yaml +++ b/.github/workflows/fine_tune.yaml @@ -25,6 +25,7 @@ jobs: - name: Install dependencies run: pip install uv + pip install -U setuptools uv pip install -r requirements.lock.txt - name: Run fine-tuning run: bash fine_tune.sh diff --git a/README.md b/README.md index f50f527..c33d03a 100644 --- a/README.md +++ b/README.md @@ -59,5 +59,5 @@ ollama run phi3-news ``` ``` -> What's the latest medical research relating to cripsr therapy? +> What are the latest top news headlines in science? ``` diff --git a/requirements.txt b/requirements.txt index 1c163a3..6fe9798 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,8 +19,8 @@ tokenizers transformers==4.41.1 torch==2.0.1 wandb -unionai==0.1.25 -flytekitplugins-envd==1.12.2 -flytekitplugins-kfpytorch==1.12.2 -flytekitplugins-huggingface==1.12.2 -flytekitplugins-deck-standard==1.12.2 +unionai==0.1.23 +flytekitplugins-envd==1.12.1rc1 +flytekitplugins-kfpytorch==1.12.1rc1 +flytekitplugins-huggingface==1.12.1rc1 +flytekitplugins-deck-standard==1.12.1rc1