simple-reranker/requirements.txt at main · cyberbobjr/simple-reranker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# --- Core Machine Learning & Deep Learning ---
torch==2.8.0
transformers==4.56.1
sentence-transformers==5.1.0
tokenizers==0.22.0
safetensors==0.6.2
numpy==2.2.6
scipy==1.15.3
scikit-learn==1.7.2
joblib==1.5.2
threadpoolctl==3.6.0

# --- Hugging Face Ecosystem (Corrected) ---
huggingface-hub==0.28.0
hf-xet==1.2.0
fsspec==2025.9.0

# --- Web & API (FastAPI) ---
fastapi==0.116.1
uvicorn==0.35.0
starlette==0.47.3
pydantic==2.11.9
pydantic_core==2.33.2
anyio==4.10.0
sniffio==1.3.1
h11==0.16.0

# --- NVIDIA / CUDA 12 Runtime ---
nvidia-cublas-cu12==12.8.4.1
nvidia-cuda-cupti-cu12==12.8.90
nvidia-cuda-nvrtc-cu12==12.8.93
nvidia-cuda-runtime-cu12==12.8.90
nvidia-cudnn-cu12==9.10.2.21
nvidia-cufft-cu12==11.3.3.83
nvidia-cufile-cu12==1.13.1.3
nvidia-curand-cu12==10.3.9.90
nvidia-cusolver-cu12==11.7.3.90
nvidia-cusparse-cu12==12.5.8.93
nvidia-cusparselt-cu12==0.7.1
nvidia-nccl-cu12==2.27.3
nvidia-nvjitlink-cu12==12.8.93
nvidia-nvtx-cu12==12.8.90
triton==3.4.0

# --- Infrastructure & Utils ---
requests==2.32.5
urllib3==2.5.0
certifi==2025.8.3
charset-normalizer==3.4.3
idna==3.10
PyYAML==6.0.2
Jinja2==3.1.6
MarkupSafe==3.0.2
pillow==11.3.0
tqdm==4.67.1
click==8.2.1
filelock==3.19.1
packaging==25.0
regex==2025.9.1
typing_extensions==4.15.0
typing-inspection==0.4.1
annotated-types==0.7.0
exceptiongroup==1.3.0
networkx==3.4.2
sympy==1.14.0
mpmath==1.3.0