torch
datasets
peft
transformers
trl
bitsandbytes
auto-gptq
optimum
einops
