torch
Pillow
transformers
accelerate>=0.26.0