einops>=0.4
einops-exts
rotary-embedding-torch
sacremosessentencepiece
torch>=1.10
torchvision
transformers[torch]
tqdm
