torch
transformers>=4.36.0
pillow
numpy