torch
transformers
Accelerate