Skip to content

Commit 1f0ea6a

Browse files
author
Ronald Tse
committed
ci: pin torch/transformers in export-fixture (transformers 5.15 tracing regression)
1 parent ccb0be0 commit 1f0ea6a

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

.github/workflows/test.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ jobs:
5252
with: { python-version: "3.11" }
5353
- run: |
5454
python -m pip install --upgrade pip
55+
pip install torch==2.12.1 transformers==5.14.1
5556
pip install -e ".[dev,train,export]"
5657
- name: IMF export + parity gate fixture tests (torch vs ORT)
5758
run: PYTHONPATH=src python -m pytest tests/test_imf_export.py tests/test_imf_parity.py -v

0 commit comments

Comments
 (0)