torch
transformers==4.42.3
einops==0.7.0
accelerate==0.32.1