pillow
torch==2.2.2
transformers
accelerate
einops