Yaz / requirements.txt
TilelliLab's picture
UX fix: requirements.txt (download instructions / safetensors / pickle-free demo)
dd920f9 verified
Raw
History Blame Contribute Delete
512 Bytes
# Yaz runtime dependencies. Tested on Python 3.11, CPU-only.
# The semantic router uses sentence-transformers (all-MiniLM-L6-v2, downloaded on first
# use); no internal/private packages are required to reproduce any result.
#
# CPU-ONLY INSTALL (recommended — avoids pulling the multi-GB CUDA stack):
# pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
torch>=2.0
numpy>=1.24
sentence-transformers>=2.2
safetensors>=0.4 # used by load_yaz.py (pickle-free weights)