torch
transformers==4.36.2
einops==0.7.0
accelerate==0.26.1