@tduncan / Nvidia-Optimization-Stack-VLLM
Code Commits Issues Pulls Cloud Agent CI/CD
main
main
in 106 lines 3.76 KB
Raw
1 -r ../common.txt
2
3 # testing
4 pytest
5 tensorizer==2.10.1
6 pytest-forked
7 pytest-asyncio
8 pytest-rerunfailures
9 pytest-shard
10 pytest-timeout
11 pytest-cov
12
13 # testing utils
14 albumentations # required for Nemotron Parse in test_common.py
15 av # required for audio_in_video tests
16 torchcodec >= 0.14 # required for torchcodec video backend tests
17 backoff # required for phi4mm test
18 blobfile # required for kimi-vl test
19 httpx
20 librosa # required for audio tests
21 vector_quantize_pytorch # required for minicpmo_26 test
22 vocos # required for minicpmo_26 test
23 peft>=0.19.1 # required for phi-4-mm test
24 pqdm
25 ray[cgraph,default]>=2.56.0,<2.57 # Ray Compiled Graph, required by pipeline parallelism tests; sharded-RDT weight transfer needs Ray Direct Transport (>=2.56)
26 sentence-transformers>=5.2.0 # required for embedding tests
27 soundfile # required for audio tests
28 jiwer # required for audio tests
29 tblib # for pickling test exceptions
30 timm >=1.0.17 # required for internvl and gemma3n-mm test
31 torch==2.13.0
32 torchaudio==2.11.0
33 torchvision==0.28.0
34 transformers_stream_generator # required for qwen-vl test
35 matplotlib # required for qwen-vl test
36 mistral_common[image,audio] >= 1.11.6 # required for voxtral test
37 num2words # required for smolvlm test
38 open_clip_torch==2.32.0 # Required for nemotron_vl test, Nemotron Parse in test_common.py
39 datamodel_code_generator>=0.64.0 # required for minicpm3 test
40 nltk>=3.10.3 # pin transitive rouge-score/lm-eval dep for known GHSAs
41 lm-eval[api]>=0.4.12 # required for model evaluation test
42 mteb[bm25s]>=2, <3 # required for mteb test
43 transformers==5.17.0
44 schemathesis>=4.0.0 # Required for openai schema test.
45 # quantization
46 buildkite-test-collector==0.1.9
47 torch-abi-audit # CI check for PyTorch stable ABI compliance
48
49 genai_perf>=0.0.8
50 tritonclient>=2.51.0
51
52 # The version of gRPC libraries should be consistent with each other
53 grpcio==1.78.0
54 grpcio-reflection==1.78.0
55
56 arctic-inference == 0.1.1; platform_machine == "x86_64" # Required for suffix decoding test
57 numba == 0.65.0 # Required for N-gram speculative decoding
58 runai-model-streamer[s3,gcs,azure]==0.15.7
59 fastsafetensors>=0.3.3
60 instanttensor>=0.1.9; platform_machine == "x86_64"
61 decord==0.6.0; platform_machine == "x86_64"
62 # terratorch is temporarily disabled while PyPI has the `lightning` package
63 # in `quarantined` status (every published terratorch version transitively
64 # requires `lightning`, so the resolver fails with "no versions of lightning").
65 # Re-enable once PyPI lifts the quarantine. Tracked in #41376.
66 # terratorch >= 1.2.2 # Required for Prithvi tests
67 imagehash # Required for Prithvi tests
68 segmentation-models-pytorch > 0.4.0 # Required for Prithvi tests
69
70 gpt-oss[eval] >= 0.0.7; python_version > '3.11'
71
72 perceptron # required for isaac test
73 kaldi-native-fbank >= 1.18.7 # required for fireredasr2 test
74 cohere_melody>=0.11.1 # required for cohere command reasoning/tool parser tests
75 cohere>=7.0.0 # required for cohere chat v2 api tests (protocol/serving import cohere.types)
76
77 # Newer versions of datasets require torchcoded, that makes the tests fail in CI because of a missing library.
78 # Older versions are in conflict with teerratorch requirements.
79 datasets>=3.3.0,<=3.6.0
80
81 openpyxl # required for perf comparison excel report
82 plotly # required for perf comparison html report
83
84 # Dependabot GHSA floors for transitive test deps (excl. dependabot.yml ignores:
85 # torch/torchvision/xformers/lm-format-enforcer/compressed-tensors/ray/lm-eval).
86 # Skipped (no patched release yet): sqlitedict, accelerate, lmcache, nltk GHSA-8mgp-746c-j5xp.
87 urllib3>=2.7.0
88 cryptography>=50.0.0
89 pyjwt>=2.14.0
90 pyasn1>=0.6.4
91 python-multipart>=0.0.31
92 anyio>=4.14.2
93 httpx2>=2.12.0
94 werkzeug>=3.1.6
95 mako>=1.3.12
96 black>=26.3.1
97 msgpack>=1.2.1
98 lxml>=6.1.0
99 pyarrow>=23.0.1
100 orjson>=3.11.6
101 hpack>=4.2.0
102 h2>=4.4.1
103 pydantic-settings>=2.14.2
104 virtualenv>=20.36.1
105 fonttools>=4.60.2
106 pygments>=2.20.0