use requirements file for tests
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user