add mistral e2e tests (#649)

* mistral e2e tests

* make sure to enable flash attention for the e2e tests

* use latest transformers full sha

* uninstall first
This commit is contained in:
Wing Lian
2023-09-29 00:22:40 -04:00
committed by GitHub
parent 9ec20777ba
commit 5b0bc48fbc
3 changed files with 210 additions and 1 deletions

View File

@@ -69,6 +69,7 @@ jobs:
- name: Install dependencies
run: |
pip3 uninstall -y transformers accelerate
pip3 install -U -e .[flash-attn]
pip3 install -r requirements-tests.txt