qwerrwe / requirements.txt
Nanobit's picture
Fix BNB OOM by pinning version
fe582df unverified
raw
history blame
227 Bytes
peft @ git+https://github.com/huggingface/peft.git
transformers @ git+https://github.com/huggingface/transformers.git
attrdict
fire
PyYAML==6.0
black
bitsandbytes==0.37.2
datasets
accelerate
sentencepiece
wandb
einops
xformers