Skip to content

Fix T4 training: fp16 instead of bf16, force single GPU (DataParallel… #163

Fix T4 training: fp16 instead of bf16, force single GPU (DataParallel…

Fix T4 training: fp16 instead of bf16, force single GPU (DataParallel… #163

Workflow file for this run

name: CI
on:
push:
branches: [main]
pull_request:
branches: [main]
jobs:
test:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.10"
cache: "pip"
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -e .
pip install pytest pytest-cov
- name: Run tests
run: |
python -m pytest tests/ -v --cov=tinydoc_vlm --cov-report=term-missing
lint:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.10"
- name: Install linting tools
run: pip install ruff
- name: Lint
run: ruff check tinydoc_vlm/ tests/