transformers
torch
accelerate
matplotlib