Skip to content

Commit 59bdfe0

Browse files
committed
Add a standalone e2e test, r6
1 parent 7858466 commit 59bdfe0

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

scripts/test-triton.sh

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -316,8 +316,13 @@ run_inductor_tests() {
316316
test -d pytorch || {
317317
git clone https://github.com/pytorch/pytorch
318318
cd pytorch
319+
319320
TRANSFORMERS_VERSION=$(cat .ci/docker/ci_commit_pins/huggingface.txt)
320-
pip install transformers==$TRANSFORMERS_VERSION
321+
git clone --recursive https://github.com/huggingface/transformers
322+
cd transformers
323+
git checkout $TRANSFORMERS_VERSION
324+
python setup.py bdist_wheel
325+
pip install dist/*.whl
321326
python -c "import transformers; print(transformers.__version__)"
322327
}
323328

0 commit comments

Comments
 (0)