mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-06 13:47:06 +00:00
45 lines
1.1 KiB
YAML
45 lines
1.1 KiB
YAML
name: Test with pytest
|
|
|
|
on:
|
|
push:
|
|
branches:
|
|
- main
|
|
- dev
|
|
- sd3
|
|
pull_request:
|
|
branches:
|
|
- main
|
|
- dev
|
|
- sd3
|
|
|
|
jobs:
|
|
build:
|
|
runs-on: ${{ matrix.os }}
|
|
strategy:
|
|
matrix:
|
|
os: [ubuntu-latest]
|
|
python-version: ["3.10"] # Python versions to test
|
|
pytorch-version: ["2.4.0"] # PyTorch versions to test
|
|
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- uses: actions/setup-python@v5
|
|
with:
|
|
python-version: ${{ matrix.python-version }}
|
|
cache: 'pip'
|
|
|
|
- name: Install and update pip, setuptools, wheel
|
|
run: |
|
|
# Setuptools, wheel for compiling some packages
|
|
python -m pip install --upgrade pip setuptools wheel
|
|
|
|
- name: Install dependencies
|
|
run: |
|
|
# Pre-install torch to pin version (requirements.txt has dependencies like transformers which requires pytorch)
|
|
pip install dadaptation==3.2 torch==${{ matrix.pytorch-version }} torchvision==0.19.0 pytest==8.3.4
|
|
pip install -r requirements.txt
|
|
|
|
- name: Test with pytest
|
|
run: pytest # See pytest.ini for configuration
|
|
|