mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-08 06:28:48 +00:00
52 lines
1.3 KiB
YAML
52 lines
1.3 KiB
YAML
name: Test with pytest
|
|
|
|
on:
|
|
push:
|
|
branches:
|
|
- main
|
|
- dev
|
|
- sd3
|
|
pull_request:
|
|
branches:
|
|
- main
|
|
- dev
|
|
- sd3
|
|
|
|
# CKV2_GHA_1: "Ensure top-level permissions are not set to write-all"
|
|
permissions: read-all
|
|
|
|
jobs:
|
|
build:
|
|
runs-on: ${{ matrix.os }}
|
|
strategy:
|
|
matrix:
|
|
os: [ubuntu-latest]
|
|
python-version: ["3.10"] # Python versions to test
|
|
pytorch-version: ["2.4.0", "2.6.0"] # PyTorch versions to test
|
|
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
with:
|
|
# https://woodruffw.github.io/zizmor/audits/#artipacked
|
|
persist-credentials: false
|
|
|
|
- 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 pytest==8.3.4
|
|
pip install -r requirements.txt
|
|
|
|
- name: Test with pytest
|
|
run: pytest # See pytest.ini for configuration
|
|
|