datasets
transformers
torch
bitsandbytes
peft
trl