Add IVON to tests

This commit is contained in:
rockerBOO
2025-06-19 14:03:00 -04:00
parent 6b810499a0
commit ba467e61be

View File

@@ -43,7 +43,7 @@ jobs:
- 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 dadaptation==3.2 torch==${{ matrix.pytorch-version }} torchvision pytest==8.3.4 git+https://github.com/rockerBOO/ivon@gradient-accumulation
pip install -r requirements.txt
- name: Test with pytest