torch
transformers>=4.38.0
Pillow
numpy
opencv-python
huggingface_hub
datasets
flash-attn
scipy
