elasticsearch-training-code / requirements_train.txt
ceperaltab's picture
Upload requirements_train.txt with huggingface_hub
4b66da1 verified
# Requirements for model training - pinned versions for compatibility
torch
transformers==4.44.2
datasets
peft
bitsandbytes
trl==0.8.6
accelerate
scipy
hf_transfer
rich