neo4j-training-code / requirements_train.txt
ceperaltab's picture
Upload requirements_train.txt with huggingface_hub
24e73e8 verified
# Requirements for model training
torch
transformers
datasets
peft
bitsandbytes
trl==0.8.6
accelerate
scipy
hf_transfer
rich