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