heaven1-base / requirements.txt
Tomas
Add initial project setup with model configuration, requirements, and upload script
58af2e6 unverified
transformers>=4.36.0
datasets>=2.14.0
torch>=2.0.0
torchvision>=0.15.0
accelerate>=0.20.0
bitsandbytes>=0.40.0
peft>=0.4.0
trl>=0.7.0
scipy>=1.10.0
numpy>=1.24.0
wandb>=0.15.0
scikit-learn>=1.2.0
tqdm>=4.65.0
jsonlines>=3.1.0
sentencepiece>=0.1.99
protobuf>=3.20.0
einops>=0.6.0
pyyaml>=6.0