RWKV-Runner/finetune/requirements.txt
2024-05-28 22:35:47 +08:00

7 lines
125 B
Plaintext

torch==2.2.0
pytorch_lightning==1.9.5
deepspeed==0.12.6
bitsandbytes==0.43.1
einops==0.8.0
triton==2.2.0
transformers==4.41.1