torch
transformers
numpy
Pillow
einops