requirements.txt

# Usage: pip install -r requirements.txt

# Base ----------------------------------------
git+https://github.com/huggingface/transformers.git@main
matplotlib>=3.2.2
Pillow>=7.1.2
torch==1.10.0
pytorch-lightning==1.9.3
opencv-python>=4.1.1