torch
transformers
accelerate