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
2023-09-29 00:22:40 -04:00
parent 9ec20777ba
commit 5b0bc48fbc
3 changed files with 210 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ torch==2.0.1
 auto-gptq
 packaging
 peft @ git+https://github.com/huggingface/peft.git
-transformers @ git+https://github.com/huggingface/transformers.git@78dd120
+transformers @ git+https://github.com/huggingface/transformers.git@5e11d72d4d0939138fbabfebe9a69d2061519547
 bitsandbytes>=0.41.1
 accelerate @ git+https://github.com/huggingface/accelerate@80da9cfb09bb3cc9f1b385cb55d6b90d025a5fd9
 deepspeed