torch
transformers
accelerate>=0.26.0