torch
Pillow
transformers>=4.41
accelerate
tokenizers
numpy