Skip to content

Commit e1fed87

Browse files
committed
update requirements
1 parent a4209da commit e1fed87

File tree

2 files changed

+15
-12
lines changed

2 files changed

+15
-12
lines changed

gpt_server/model_worker/embedding_vllm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def __init__(
4646
)
4747
tensor_parallel_size = int(os.getenv("num_gpus", "1"))
4848
max_model_len = os.getenv("max_model_len", None)
49-
gpu_memory_utilization = float(os.getenv("gpu_memory_utilization", 0.8))
49+
gpu_memory_utilization = float(os.getenv("gpu_memory_utilization", 0.6))
5050
enable_prefix_caching = bool(os.getenv("enable_prefix_caching", False))
5151

5252
self.mode = get_embedding_mode(model_path=model_path)

requirements.txt

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ distro==1.9.0
182182
# anthropic
183183
# openai
184184
# posthog
185-
dnspython==2.7.0
185+
dnspython==2.8.0
186186
# via email-validator
187187
edge-tts==7.2.3
188188
# via gpt-server (pyproject.toml)
@@ -381,7 +381,7 @@ jsonschema==4.25.1
381381
# mistral-common
382382
# outlines
383383
# ray
384-
jsonschema-specifications==2025.4.1
384+
jsonschema-specifications==2025.9.1
385385
# via jsonschema
386386
kaldiio==2.18.1
387387
# via funasr
@@ -412,7 +412,7 @@ langchain-text-splitters==0.3.11
412412
# via langchain
413413
langdetect==1.0.9
414414
# via evalscope
415-
langsmith==0.4.25
415+
langsmith==0.4.26
416416
# via
417417
# langchain
418418
# langchain-community
@@ -499,7 +499,7 @@ multiprocess==0.70.16
499499
# via datasets
500500
mypy-extensions==1.1.0
501501
# via typing-inspect
502-
narwhals==2.3.0
502+
narwhals==2.4.0
503503
# via altair
504504
nest-asyncio==1.6.0
505505
# via
@@ -563,6 +563,7 @@ numpy==1.26.4
563563
# pydeck
564564
# pyloudnorm
565565
# pytorch-wpe
566+
# pytrec-eval-terrier
566567
# ragas
567568
# rouge-score
568569
# sacrebleu
@@ -619,7 +620,7 @@ nvidia-cusparse-cu12==12.5.8.93
619620
# torch
620621
nvidia-cusparselt-cu12==0.7.1
621622
# via torch
622-
nvidia-ml-py==12.575.51
623+
nvidia-ml-py==13.580.65
623624
# via pynvml
624625
nvidia-nccl-cu12==2.27.3
625626
# via
@@ -631,7 +632,7 @@ nvidia-nvjitlink-cu12==12.8.93
631632
# nvidia-cusolver-cu12
632633
# nvidia-cusparse-cu12
633634
# torch
634-
nvidia-nvshmem-cu12==3.3.24
635+
nvidia-nvshmem-cu12==3.4.5
635636
# via flashinfer-python
636637
nvidia-nvtx-cu12==12.8.90
637638
# via torch
@@ -751,7 +752,7 @@ pooch==1.8.2
751752
# via librosa
752753
portalocker==3.2.0
753754
# via sacrebleu
754-
posthog==6.7.3
755+
posthog==6.7.4
755756
# via infinity-emb
756757
prometheus-client==0.22.1
757758
# via
@@ -855,7 +856,7 @@ pynini==2.1.6
855856
# via wetextprocessing
856857
pynndescent==0.5.13
857858
# via umap-learn
858-
pynvml==12.0.0
859+
pynvml==13.0.1
859860
# via
860861
# flashinfer-python
861862
# lmdeploy
@@ -883,7 +884,7 @@ python-multipart==0.0.20
883884
# sglang
884885
pytorch-wpe==0.0.1
885886
# via funasr
886-
pytrec-eval-terrier==0.5.8
887+
pytrec-eval-terrier==0.5.9
887888
# via mteb
888889
pytz==2025.2
889890
# via pandas
@@ -1013,6 +1014,7 @@ scipy==1.16.1
10131014
# mteb
10141015
# pyloudnorm
10151016
# pynndescent
1017+
# pytrec-eval-terrier
10161018
# scikit-learn
10171019
# sentence-transformers
10181020
# sglang
@@ -1084,7 +1086,7 @@ soundfile==0.13.1
10841086
# librosa
10851087
# mistral-common
10861088
# sglang
1087-
soxr==0.5.0.post1
1089+
soxr==1.0.0
10881090
# via
10891091
# flashtts
10901092
# librosa
@@ -1311,7 +1313,8 @@ uvloop==0.21.0
13111313
# via
13121314
# sglang
13131315
# uvicorn
1314-
vllm @ git+https://gitee.com/liuyu_1997/vllm.git@006e7a34aeb3e905ca4131a3251fe079f0511e2f
1316+
--find-links https://wheels.vllm.ai/006e7a34aeb3e905ca4131a3251fe079f0511e2f
1317+
vllm
13151318
# via gpt-server (pyproject.toml)
13161319
watchdog==5.0.3
13171320
# via streamlit

0 commit comments

Comments
 (0)