| # 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 | |