torch
transformers
datasets
accelerate
bitsandbytes
wandb
peft
deepspeed