RWKV-Runner/finetune/requirements.txt

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
sync https://github.com/JL-er/RWKV-PEFT 2024-05-28 22:35:47 +08:00			`torch==2.2.0`
lora finetune (need to be refactored) 2023-07-03 17:41:47 +08:00			`pytorch_lightning==1.9.5`
improve fine-tune performance 2024-02-04 19:33:32 +08:00			`deepspeed==0.12.6`
sync https://github.com/JL-er/RWKV-PEFT 2024-05-28 22:35:47 +08:00			`bitsandbytes==0.43.1`
			`einops==0.8.0`
			`triton==2.2.0`
			`transformers==4.41.1`