# Python >= 3.10. Install torch for your CUDA version first, e.g.:
#   pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
torch>=2.4
torchvision
transformers>=5.20      # needed for google/embeddinggemma-2 (model_type embedding_gemma2); tested with 5.20.1
sentence-transformers>=6.1   # tested with 6.1.0
huggingface_hub
datasets
pandas
pyarrow
numpy
pyyaml
pillow
sentencepiece
open_clip_torch         # only if image_encoder.backend=open_clip