qwerrwe / requirements-tests.txt
winglian's picture
fix path and streamline pip installs
403af0b
raw
history blame
361 Bytes
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