-
Notifications
You must be signed in to change notification settings - Fork 79
38 lines (31 loc) · 978 Bytes
/
ci-cd.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
name: CI/CD
on: pull_request
jobs:
test:
environment: CI
name: Deploy
runs-on: ubuntu-latest
strategy:
matrix:
config: ["codellama", "llama-2", "mistral"]
env:
MODAL_TOKEN_ID: ${{ secrets.MODAL_TOKEN_ID }}
MODAL_TOKEN_SECRET: ${{ secrets.MODAL_TOKEN_SECRET }}
MODAL_ENVIRONMENT: ci-cd
steps:
- name: Checkout Repository
uses: actions/checkout@v4
- name: Install Python
uses: actions/setup-python@v5
with:
python-version: "3.12"
- name: Install Modal
run: |
python -m pip install --upgrade pip
pip install modal pyyaml
- name: Prep config and data for CI
run: |
python ci/prep_for_ci.py --config=config/{{ matrix.config }}.yml --data=data/sqlqa.jsonl
- name: Run training job on Modal
run: |
GPU_MEM=40 modal run src.train --config=config/{{ amtrix.confg }}.yml --data=data/sqlqa.jsonl