Skip to content

RL training of Llama 3.1 8b on multi-gpus #401

RL training of Llama 3.1 8b on multi-gpus

RL training of Llama 3.1 8b on multi-gpus #401

Workflow file for this run

# This workflow will install Python dependencies, run tests and lint with a single version of Python
# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python
name: Python Tests
on:
push:
branches: ["main"]
pull_request:
branches: ["main"]
permissions:
contents: read
jobs:
test:
runs-on: ubuntu-22.04
steps:
- uses: actions/checkout@v4
- name: Set up Python 3.10
uses: actions/setup-python@v5
with:
python-version: "3.10"
cache: "pip"
- name: Install dependencies and package
run: |
python -m pip install --upgrade pip
pip install -r requirements.dev.txt -r requirements.txt -r requirements.finetune.txt -r requirements.converters.txt
pip install -e .
- name: Lint critical errors with flake8
run: |
# stop the build if there are Python syntax errors or undefined names
flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
- name: Run fast tests with pytest
run: |
HF_TOKEN="${{ secrets.HF_HUB_TOKEN }}" pytest -m "not slow" tests