torch
transformers==5.5.4
accelerate