File size: 475 Bytes
989d4ef | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | # 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 |