use requirements file for tests

This commit is contained in:
Wing Lian
2023-05-27 12:17:46 -04:00
parent 72b6ca0d9f
commit a653392287
2 changed files with 2 additions and 14 deletions

View File

@@ -1,13 +1 @@
pytest
# this is a stripped down version so it doesn't try to install some of the things like xformers or others that may require compilation
peft @ git+https://github.com/huggingface/peft.git
transformers @ git+https://github.com/huggingface/transformers.git
bitsandbytes>=0.39.0
attrdict
fire
PyYAML==6.0
black
datasets
accelerate>=0.19.0
sentencepiece
wandb