torch
transformers
accelerate
pillow
einops
spaces