UX fix: requirements.txt (download instructions / safetensors / pickle-free demo)
Browse files- requirements.txt +4 -0
requirements.txt
CHANGED
|
@@ -1,6 +1,10 @@
|
|
| 1 |
# Yaz runtime dependencies. Tested on Python 3.11, CPU-only.
|
| 2 |
# The semantic router uses sentence-transformers (all-MiniLM-L6-v2, downloaded on first
|
| 3 |
# use); no internal/private packages are required to reproduce any result.
|
|
|
|
|
|
|
|
|
|
| 4 |
torch>=2.0
|
| 5 |
numpy>=1.24
|
| 6 |
sentence-transformers>=2.2
|
|
|
|
|
|
| 1 |
# Yaz runtime dependencies. Tested on Python 3.11, CPU-only.
|
| 2 |
# The semantic router uses sentence-transformers (all-MiniLM-L6-v2, downloaded on first
|
| 3 |
# use); no internal/private packages are required to reproduce any result.
|
| 4 |
+
#
|
| 5 |
+
# CPU-ONLY INSTALL (recommended — avoids pulling the multi-GB CUDA stack):
|
| 6 |
+
# pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
|
| 7 |
torch>=2.0
|
| 8 |
numpy>=1.24
|
| 9 |
sentence-transformers>=2.2
|
| 10 |
+
safetensors>=0.4 # used by load_yaz.py (pickle-free weights)
|