torch==2.0.1
transformers==4.31.0
einops==0.6.0
Pillow==9.2.0