rlm-training-test / requirements.txt
mindchain's picture
Add requirements.txt
2a7f875 verified
raw
history blame contribute delete
116 Bytes
torch>=2.0.0
transformers>=4.51.0
datasets>=2.14.0
accelerate>=0.27.0
trl>=0.12.0
peft>=0.10.0
bitsandbytes>=0.43.0