torch
transformers==4.31.0
tokenizers
datasets==2.15.0
peft
wandb
loguru
nvitop
lion-pytorch
matplotlib
bitsandbytes
scipy
scikit-learn
evaluate
deepspeed
