torch
peft
accelerate
transformers
einops