@tduncan / Nvidia-Optimization-Stack-VLLM Mirror · read-only / commits / db01535e2b
Fork 0

A high-throughput and memory-efficient inference and serving engine for LLMs

Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git

Synced Nov 5, 22:15 UTC

[ROCm][CI] Add uv pip compile workflow for rocm-test.txt lockfile (#37930)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>

Andreas Karatzas committed Mar 26, 2026 at 12:44 UTC db01535e2bd5641e8cd7f88e5febf1d2a26bfbce
4 files changed +1478 -98
.pre-commit-config.yaml
+36 -1
@@ -36,11 +36,46 @@ repos:
36 hooks:
37 - id: actionlint
38 - repo: https://github.com/astral-sh/uv-pre-commit
39 - rev: 0.9.1
39 + rev: 0.11.1
40 hooks:
41 - id: pip-compile
42 args: [requirements/test.in, -o, requirements/test.txt, --index-strategy, unsafe-best-match, --torch-backend, cu129, --python-platform, x86_64-manylinux_2_28, --python-version, "3.12"]
43 files: ^requirements/test\.(in|txt)$
44 + - id: pip-compile
45 + alias: pip-compile-rocm
46 + name: pip-compile-rocm
47 + args: [
48 + requirements/rocm-test.in, -o, requirements/rocm-test.txt,
49 + --index-strategy, unsafe-best-match,
50 + -c, requirements/rocm.txt,
51 + --python-platform, x86_64-manylinux_2_28,
52 + --python-version, "3.12",
53 + # Exclude torch and CUDA/NVIDIA packages
54 + --no-emit-package, torch,
55 + --no-emit-package, torchvision,
56 + --no-emit-package, torchaudio,
57 + --no-emit-package, triton,
58 + --no-emit-package, cuda-bindings,
59 + --no-emit-package, cuda-pathfinder,
60 + --no-emit-package, cuda-toolkit,
61 + --no-emit-package, cupy-cuda12x,
62 + --no-emit-package, nvidia-cublas,
63 + --no-emit-package, nvidia-cuda-cupti,
64 + --no-emit-package, nvidia-cuda-nvrtc,
65 + --no-emit-package, nvidia-cuda-runtime,
66 + --no-emit-package, nvidia-cudnn-cu13,
67 + --no-emit-package, nvidia-cufft,
68 + --no-emit-package, nvidia-cufile,
69 + --no-emit-package, nvidia-curand,
70 + --no-emit-package, nvidia-cusolver,
71 + --no-emit-package, nvidia-cusparse,
72 + --no-emit-package, nvidia-cusparselt-cu13,
73 + --no-emit-package, nvidia-nccl-cu13,
74 + --no-emit-package, nvidia-nvjitlink,
75 + --no-emit-package, nvidia-nvshmem-cu13,
76 + --no-emit-package, nvidia-nvtx,
77 + ]
78 + files: ^requirements/rocm-test\.(in|txt)$
79 - repo: local
80 hooks:
81 - id: format-torch-nightly-test
docker/Dockerfile.rocm
+5
@@ -329,6 +329,11 @@ RUN --mount=type=bind,from=export_vllm,src=/,target=/install \
329 && pip uninstall -y vllm \
330 && uv pip install --system *.whl
331
332 +# Verify that PyTorch is the ROCm build, not CUDA
333 +RUN python3 -c "import torch; assert torch.version.hip is not None, \
334 + f'Expected ROCm PyTorch but got CUDA (torch.version.cuda={torch.version.cuda}, torch.version.hip={torch.version.hip})'; \
335 + print(f'Verified: PyTorch {torch.__version__} with ROCm (HIP {torch.version.hip})')"
336 +
337 # Install RIXL wheel
338 RUN --mount=type=bind,from=build_rixl,src=/app/install,target=/rixl_install \
339 uv pip install --system /rixl_install/*.whl
requirements/rocm-test.in new
+83
@@ -0,0 +1,83 @@
1 +# testing
2 +pytest
3 +tensorizer==2.10.1
4 +pytest-forked
5 +pytest-asyncio
6 +pytest-rerunfailures
7 +pytest-shard
8 +pytest-timeout
9 +pytest-cov
10 +
11 +# testing utils
12 +albumentations # required for Nemotron Parse in test_common.py
13 +av # required for audio_in_video tests
14 +backoff # required for phi4mm test
15 +blobfile # required for kimi-vl test
16 +einops # required for MPT, qwen-vl
17 +httpx
18 +librosa # required for audio tests
19 +vector_quantize_pytorch # required for minicpmo_26 test
20 +vocos # required for minicpmo_26 test
21 +peft>=0.15.0 # required for phi-4-mm test
22 +pqdm
23 +ray[cgraph,default]>=2.48.0 # Ray Compiled Graph, required by pipeline parallelism tests
24 +resampy # required for audio tests
25 +sentence-transformers>=5.2.0 # required for embedding tests
26 +soundfile # required for audio tests
27 +jiwer # required for audio tests
28 +tblib # for pickling test exceptions
29 +timm>=1.0.17 # required for internvl and gemma3n-mm test
30 +transformers_stream_generator # required for qwen-vl test
31 +matplotlib # required for qwen-vl test
32 +mistral_common[image,audio]>=1.10.0 # required for voxtral test
33 +num2words # required for smolvlm test
34 +open_clip_torch==2.32.0 # Required for nemotron_vl test, Nemotron Parse in test_common.py
35 +opencv-python-headless>=4.13.0 # required for video test
36 +datamodel_code_generator # required for minicpm3 test
37 +lm-eval[api]>=0.4.11 # required for model evaluation test
38 +mteb[bm25s]>=2, <3 # required for mteb test
39 +transformers==4.57.5
40 +tokenizers==0.22.0
41 +schemathesis>=3.39.15 # Required for openai schema test
42 +# quantization
43 +bitsandbytes==0.49.2
44 +buildkite-test-collector==0.1.9
45 +
46 +genai_perf>=0.0.8
47 +tritonclient>=2.51.0
48 +
49 +# The version of gRPC libraries should be consistent with each other
50 +grpcio==1.78.0
51 +grpcio-reflection==1.78.0
52 +
53 +arctic-inference==0.1.1 # Required for suffix decoding test
54 +numba==0.61.2 # Required for N-gram speculative decoding
55 +numpy
56 +runai-model-streamer[s3,gcs,azure]==0.15.7
57 +fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage
58 +instanttensor>=0.1.5
59 +pydantic>=2.12 # 2.11 leads to error on python 3.13
60 +decord==0.6.0
61 +
62 +# Prithvi tests
63 +terratorch>=1.2.2
64 +imagehash # Required for Prithvi tests
65 +segmentation-models-pytorch>0.4.0 # Required for Prithvi tests
66 +
67 +gpt-oss>=0.0.7; python_version > '3.11'
68 +
69 +perceptron # required for isaac test
70 +kaldi-native-fbank>=1.18.7 # required for fireredasr2 test
71 +
72 +# Newer versions of datasets require torchcoded, that makes the tests fail in CI because of a missing library.
73 +# Older versions are in conflict with terratorch requirements.
74 +datasets>=3.3.0,<=3.6.0
75 +
76 +openpyxl # required for perf comparison excel report
77 +plotly # required for perf comparison html report
78 +
79 +# ROCm-specific extras (not in CUDA test.in)
80 +rapidfuzz
81 +torchgeo==0.7.0
82 +multiprocess==0.70.16
83 +huggingface-hub==0.36.2
requirements/rocm-test.txt
+1354 -97
@@ -1,116 +1,1373 @@
1 -# Common dependencies
2 --r common.txt
3 -
4 -# Test infrastructure
5 -tblib==3.1.0
6 -pytest==8.3.5
7 -pytest-asyncio==0.24.0
8 -pytest-timeout==2.3.1
9 -pytest-cov==6.3.0
10 -pytest-forked==1.6.0
11 -pytest-rerunfailures==14.0
12 -pytest-shard==0.1.2
13 -
14 -# Async/HTTP dependencies
15 -anyio==4.6.2.post1
16 - # via httpx, starlette
1 +# This file was autogenerated by uv via the following command:
2 +# uv pip compile requirements/rocm-test.in -o requirements/rocm-test.txt --index-strategy unsafe-best-match -c requirements/rocm.txt --python-platform x86_64-manylinux_2_28 --python-version 3.12 --no-emit-package torch --no-emit-package torchvision --no-emit-package torchaudio --no-emit-package triton --no-emit-package cuda-bindings --no-emit-package cuda-pathfinder --no-emit-package cuda-toolkit --no-emit-package cupy-cuda12x --no-emit-package nvidia-cublas --no-emit-package nvidia-cuda-cupti --no-emit-package nvidia-cuda-nvrtc --no-emit-package nvidia-cuda-runtime --no-emit-package nvidia-cudnn-cu13 --no-emit-package nvidia-cufft --no-emit-package nvidia-cufile --no-emit-package nvidia-curand --no-emit-package nvidia-cusolver --no-emit-package nvidia-cusparse --no-emit-package nvidia-cusparselt-cu13 --no-emit-package nvidia-nccl-cu13 --no-emit-package nvidia-nvjitlink --no-emit-package nvidia-nvshmem-cu13 --no-emit-package nvidia-nvtx
3 +absl-py==2.4.0
4 + # via
5 + # rouge-score
6 + # tensorboard
7 +accelerate==1.13.0
8 + # via peft
9 +aenum==3.1.17
10 + # via lightly
11 +affine==2.4.0
12 + # via rasterio
13 +aiohappyeyeballs==2.6.1
14 + # via aiohttp
15 aiohttp==3.13.3
18 - # via gpt-oss
19 -httpx==0.27.2
20 - # HTTP testing
21 -
22 -# Audio processing dependencies
16 + # via
17 + # -c requirements/common.txt
18 + # aiohttp-cors
19 + # fsspec
20 + # gpt-oss
21 + # lm-eval
22 + # ray
23 +aiohttp-cors==0.8.1
24 + # via ray
25 +aiosignal==1.4.0
26 + # via aiohttp
27 +albucore==0.1.2
28 + # via terratorch
29 +albumentations==1.4.6
30 + # via
31 + # -r requirements/rocm-test.in
32 + # terratorch
33 +alembic==1.18.4
34 + # via optuna
35 +annotated-doc==0.0.4
36 + # via
37 + # fastapi
38 + # typer
39 +annotated-types==0.7.0
40 + # via pydantic
41 +antlr4-python3-runtime==4.9.3
42 + # via
43 + # hydra-core
44 + # omegaconf
45 +anyio==4.6.2.post1
46 + # via
47 + # httpx
48 + # starlette
49 +arctic-inference==0.1.1
50 + # via -r requirements/rocm-test.in
51 +argcomplete==3.6.3
52 + # via datamodel-code-generator
53 +arrow==1.4.0
54 + # via isoduration
55 +attrs==26.1.0
56 + # via
57 + # aiohttp
58 + # fiona
59 + # jsonlines
60 + # jsonschema
61 + # pytest-subtests
62 + # rasterio
63 + # referencing
64 audioread==3.0.1
65 # via librosa
66 +av==16.1.0
67 + # via -r requirements/rocm-test.in
68 +azure-core==1.39.0
69 + # via
70 + # azure-identity
71 + # azure-storage-blob
72 +azure-identity==1.25.3
73 + # via runai-model-streamer-azure
74 +azure-storage-blob==12.28.0
75 + # via runai-model-streamer-azure
76 +backoff==2.2.1
77 + # via
78 + # -r requirements/rocm-test.in
79 + # schemathesis
80 +bitsandbytes==0.49.2
81 + # via
82 + # -r requirements/rocm-test.in
83 + # lightning
84 +black==26.3.1
85 + # via datamodel-code-generator
86 +blobfile==3.0.0
87 + # via -r requirements/rocm-test.in
88 +bm25s==0.2.13
89 + # via mteb
90 +boto3==1.42.74
91 + # via
92 + # runai-model-streamer-s3
93 + # tensorizer
94 +botocore==1.42.74
95 + # via
96 + # boto3
97 + # s3transfer
98 +bounded-pool-executor==0.0.3
99 + # via pqdm
100 +buildkite-test-collector==0.1.9
101 + # via -r requirements/rocm-test.in
102 +certifi==2026.2.25
103 + # via
104 + # fiona
105 + # httpcore
106 + # httpx
107 + # lightly
108 + # pyogrio
109 + # pyproj
110 + # rasterio
111 + # requests
112 + # sentry-sdk
113 cffi==1.17.1
26 - # via soundfile
114 + # via
115 + # cryptography
116 + # soundfile
117 +chardet==5.2.0
118 + # via mbstrdecoder
119 +charset-normalizer==3.4.6
120 + # via requests
121 +choreographer==1.2.1
122 + # via kaleido
123 +chz==0.4.0
124 + # via gpt-oss
125 +click==8.3.1
126 + # via
127 + # black
128 + # click-plugins
129 + # cligj
130 + # fiona
131 + # jiwer
132 + # nltk
133 + # rasterio
134 + # ray
135 + # schemathesis
136 + # typer
137 + # uvicorn
138 + # wandb
139 +click-plugins==1.1.1.2
140 + # via fiona
141 +cligj==0.7.2
142 + # via
143 + # fiona
144 + # rasterio
145 +colorama==0.4.6
146 + # via
147 + # perceptron
148 + # sacrebleu
149 + # schemathesis
150 +colorful==0.5.8
151 + # via ray
152 +colorlog==6.10.1
153 + # via optuna
154 +contourpy==1.3.3
155 + # via matplotlib
156 +coverage==7.13.5
157 + # via pytest-cov
158 +cramjam==2.11.0
159 + # via fastparquet
160 +cryptography==46.0.0
161 + # via
162 + # azure-identity
163 + # azure-storage-blob
164 + # google-auth
165 + # msal
166 + # pyjwt
167 +cycler==0.12.1
168 + # via matplotlib
169 +datamodel-code-generator==0.55.0
170 + # via -r requirements/rocm-test.in
171 +dataproperty==1.1.0
172 + # via
173 + # pytablewriter
174 + # tabledata
175 +datasets==3.6.0
176 + # via
177 + # -r requirements/rocm-test.in
178 + # evaluate
179 + # lm-eval
180 + # mteb
181 decorator==5.2.1
182 # via librosa
183 +decord==0.6.0
184 + # via -r requirements/rocm-test.in
185 +diffusers==0.37.0
186 + # via terratorch
187 +dill==0.3.8
188 + # via
189 + # datasets
190 + # evaluate
191 + # lm-eval
192 + # multiprocess
193 +distlib==0.4.0
194 + # via virtualenv
195 +docker==7.1.0
196 + # via gpt-oss
197 +docopt==0.6.2
198 + # via num2words
199 +docstring-parser==0.17.0
200 + # via jsonargparse
201 +einops==0.8.2
202 + # via
203 + # -r requirements/rocm-test.in
204 + # encodec
205 + # terratorch
206 + # torchgeo
207 + # vector-quantize-pytorch
208 + # vocos
209 +einx==0.4.2
210 + # via vector-quantize-pytorch
211 +encodec==0.1.1
212 + # via vocos
213 +et-xmlfile==2.0.0
214 + # via openpyxl
215 +evaluate==0.4.6
216 + # via lm-eval
217 +fastapi==0.135.2
218 + # via
219 + # -c requirements/common.txt
220 + # gpt-oss
221 +fastparquet==2026.3.0
222 + # via genai-perf
223 +fastsafetensors==0.2.2
224 + # via -r requirements/rocm-test.in
225 +filelock==3.25.2
226 + # via
227 + # -c requirements/common.txt
228 + # blobfile
229 + # datasets
230 + # diffusers
231 + # huggingface-hub
232 + # python-discovery
233 + # ray
234 + # torch
235 + # transformers
236 + # virtualenv
237 +fiona==1.10.1
238 + # via torchgeo
239 +fonttools==4.62.1
240 + # via matplotlib
241 +fqdn==1.5.1
242 + # via jsonschema
243 +frozendict==2.4.7
244 + # via einx
245 +frozenlist==1.8.0
246 + # via
247 + # aiohttp
248 + # aiosignal
249 +fsspec==2025.3.0
250 + # via
251 + # datasets
252 + # evaluate
253 + # fastparquet
254 + # huggingface-hub
255 + # lightning
256 + # pytorch-lightning
257 + # tacoreader
258 + # torch
259 +ftfy==6.3.1
260 + # via open-clip-torch
261 +genai-perf==0.0.16
262 + # via -r requirements/rocm-test.in
263 +genson==1.3.0
264 + # via datamodel-code-generator
265 +geopandas==1.1.3
266 + # via terratorch
267 +gitdb==4.0.12
268 + # via gitpython
269 +gitpython==3.1.46
270 + # via wandb
271 +google-api-core==2.30.0
272 + # via
273 + # google-cloud-core
274 + # google-cloud-storage
275 + # opencensus
276 +google-auth==2.49.1
277 + # via
278 + # google-api-core
279 + # google-cloud-core
280 + # google-cloud-storage
281 + # runai-model-streamer-gcs
282 +google-cloud-core==2.5.0
283 + # via google-cloud-storage
284 +google-cloud-storage==3.10.1
285 + # via runai-model-streamer-gcs
286 +google-crc32c==1.8.0
287 + # via
288 + # google-cloud-storage
289 + # google-resumable-media
290 +google-resumable-media==2.8.0
291 + # via google-cloud-storage
292 +googleapis-common-protos==1.73.0
293 + # via google-api-core
294 +gpt-oss==0.0.8
295 + # via -r requirements/rocm-test.in
296 +graphql-core==3.2.8
297 + # via hypothesis-graphql
298 +greenlet==3.3.2
299 + # via sqlalchemy
300 +grpcio==1.78.0
301 + # via
302 + # -c requirements/rocm.txt
303 + # -r requirements/rocm-test.in
304 + # grpcio-reflection
305 + # ray
306 + # tensorboard
307 +grpcio-reflection==1.78.0
308 + # via
309 + # -c requirements/rocm.txt
310 + # -r requirements/rocm-test.in
311 +h11==0.16.0
312 + # via
313 + # httpcore
314 + # uvicorn
315 +h2==4.3.0
316 + # via httpx
317 +h5py==3.16.0
318 + # via terratorch
319 +harfile==0.4.0
320 + # via schemathesis
321 +hf-xet==1.4.2
322 + # via huggingface-hub
323 +hiredis==3.3.1
324 + # via tensorizer
325 +hpack==4.1.0
326 + # via h2
327 +html2text==2025.4.15
328 + # via gpt-oss
329 +httpcore==1.0.9
330 + # via httpx
331 +httpx==0.27.2
332 + # via
333 + # -r requirements/rocm-test.in
334 + # diffusers
335 + # perceptron
336 + # schemathesis
337 +huggingface-hub==0.36.2
338 + # via
339 + # -r requirements/rocm-test.in
340 + # accelerate
341 + # datasets
342 + # diffusers
343 + # evaluate
344 + # open-clip-torch
345 + # peft
346 + # segmentation-models-pytorch
347 + # sentence-transformers
348 + # terratorch
349 + # timm
350 + # tokenizers
351 + # transformers
352 + # vocos
353 +humanize==4.15.0
354 + # via runai-model-streamer
355 +hydra-core==1.3.2
356 + # via
357 + # lightly
358 + # lightning
359 +hyperframe==6.1.0
360 + # via h2
361 +hypothesis==6.151.9
362 + # via
363 + # hypothesis-graphql
364 + # hypothesis-jsonschema
365 + # schemathesis
366 +hypothesis-graphql==0.12.0
367 + # via schemathesis
368 +hypothesis-jsonschema==0.23.1
369 + # via schemathesis
370 +idna==3.11
371 + # via
372 + # anyio
373 + # httpx
374 + # jsonschema
375 + # requests
376 + # yarl
377 +imagehash==4.3.2
378 + # via -r requirements/rocm-test.in
379 +imageio==2.37.3
380 + # via scikit-image
381 +importlib-metadata==8.7.1
382 + # via
383 + # diffusers
384 + # opentelemetry-api
385 +importlib-resources==6.5.2
386 + # via typeshed-client
387 +inflect==7.5.0
388 + # via datamodel-code-generator
389 +iniconfig==2.3.0
390 + # via pytest
391 +instanttensor==0.1.6
392 + # via -r requirements/rocm-test.in
393 +isodate==0.7.2
394 + # via azure-storage-blob
395 +isoduration==20.11.0
396 + # via jsonschema
397 +isort==8.0.1
398 + # via datamodel-code-generator
399 +jinja2==3.1.6
400 + # via
401 + # datamodel-code-generator
402 + # genai-perf
403 + # lm-eval
404 + # torch
405 +jiwer==4.0.0
406 + # via -r requirements/rocm-test.in
407 +jmespath==1.1.0
408 + # via
409 + # boto3
410 + # botocore
411 +joblib==1.5.3
412 + # via
413 + # librosa
414 + # nltk
415 + # scikit-learn
416 +jsonargparse==4.47.0
417 + # via
418 + # lightning
419 + # terratorch
420 +jsonlines==4.0.0
421 + # via lm-eval
422 +jsonnet==0.21.0
423 + # via jsonargparse
424 +jsonpointer==3.1.0
425 + # via jsonschema
426 +jsonschema==4.26.0
427 + # via
428 + # hypothesis-jsonschema
429 + # mistral-common
430 + # ray
431 + # schemathesis
432 +jsonschema-specifications==2025.9.1
433 + # via jsonschema
434 +junit-xml==1.9
435 + # via schemathesis
436 +kaldi-native-fbank==1.22.3
437 + # via -r requirements/rocm-test.in
438 +kaleido==1.0.0
439 + # via genai-perf
440 +kiwisolver==1.5.0
441 + # via matplotlib
442 +kornia==0.8.2
443 + # via torchgeo
444 +kornia-rs==0.1.10
445 + # via kornia
446 lazy-loader==0.4
30 - # via librosa
447 + # via
448 + # librosa
449 + # scikit-image
450 +libnacl==2.1.0
451 + # via tensorizer
452 +librosa==0.10.2.post1
453 + # via -r requirements/rocm-test.in
454 +lightly==1.5.22
455 + # via
456 + # terratorch
457 + # torchgeo
458 +lightly-utils==0.0.2
459 + # via lightly
460 +lightning==2.6.1
461 + # via
462 + # terratorch
463 + # torchgeo
464 +lightning-utilities==0.15.3
465 + # via
466 + # lightning
467 + # pytorch-lightning
468 + # torchmetrics
469 +llvmlite==0.44.0
470 + # via numba
471 +lm-eval==0.4.11
472 + # via -r requirements/rocm-test.in
473 +logistro==2.0.1
474 + # via
475 + # choreographer
476 + # kaleido
477 +lxml==6.0.2
478 + # via
479 + # blobfile
480 + # gpt-oss
481 + # sacrebleu
482 +mako==1.3.10
483 + # via alembic
484 +markdown==3.10.2
485 + # via tensorboard
486 +markdown-it-py==4.0.0
487 + # via rich
488 +markupsafe==3.0.3
489 + # via
490 + # jinja2
491 + # mako
492 + # werkzeug
493 +matplotlib==3.10.8
494 + # via
495 + # -r requirements/rocm-test.in
496 + # lightning
497 + # torchgeo
498 +mbstrdecoder==1.1.4
499 + # via
500 + # dataproperty
501 + # pytablewriter
502 + # typepy
503 +mdurl==0.1.2
504 + # via markdown-it-py
505 +mistral-common==1.10.0
506 + # via
507 + # -c requirements/common.txt
508 + # -r requirements/rocm-test.in
509 +more-itertools==10.8.0
510 + # via
511 + # inflect
512 + # lm-eval
513 +mpmath==1.3.0
514 + # via sympy
515 +msal==1.35.1
516 + # via
517 + # azure-identity
518 + # msal-extensions
519 +msal-extensions==1.3.1
520 + # via azure-identity
521 +msgpack==1.1.2
522 + # via
523 + # librosa
524 + # ray
525 +mteb==2.11.5
526 + # via -r requirements/rocm-test.in
527 +multidict==6.7.1
528 + # via
529 + # aiohttp
530 + # yarl
531 +multiprocess==0.70.16
532 + # via
533 + # -r requirements/rocm-test.in
534 + # datasets
535 + # evaluate
536 +mypy-extensions==1.1.0
537 + # via black
538 +narwhals==2.18.0
539 + # via plotly
540 +networkx==3.6.1
541 + # via
542 + # scikit-image
543 + # torch
544 +nltk==3.9.3
545 + # via rouge-score
546 +num2words==0.5.14
547 + # via -r requirements/rocm-test.in
548 +numba==0.61.2
549 + # via
550 + # -c requirements/rocm.txt
551 + # -r requirements/rocm-test.in
552 + # librosa
553 + # resampy
554 +numkong==7.1.1
555 + # via albucore
556 +numpy==2.2.6
557 + # via
558 + # -r requirements/rocm-test.in
559 + # accelerate
560 + # albucore
561 + # albumentations
562 + # bitsandbytes
563 + # bm25s
564 + # contourpy
565 + # cupy-cuda12x
566 + # datasets
567 + # decord
568 + # diffusers
569 + # einx
570 + # encodec
571 + # evaluate
572 + # fastparquet
573 + # genai-perf
574 + # geopandas
575 + # h5py
576 + # imagehash
577 + # imageio
578 + # librosa
579 + # lightly
580 + # lightly-utils
581 + # lm-eval
582 + # matplotlib
583 + # mistral-common
584 + # mteb
585 + # numba
586 + # opencv-python-headless
587 + # optuna
588 + # pandas
589 + # patsy
590 + # peft
591 + # perceptron
592 + # pycocotools
593 + # pyogrio
594 + # pytrec-eval-terrier
595 + # pywavelets
596 + # rasterio
597 + # resampy
598 + # rioxarray
599 + # rouge-score
600 + # runai-model-streamer
601 + # sacrebleu
602 + # scikit-image
603 + # scikit-learn
604 + # scipy
605 + # segmentation-models-pytorch
606 + # sentence-transformers
607 + # shapely
608 + # soundfile
609 + # soxr
610 + # statsmodels
611 + # tensorboard
612 + # tensorboardx
613 + # tensorizer
614 + # terratorch
615 + # tifffile
616 + # torchgeo
617 + # torchmetrics
618 + # torchvision
619 + # transformers
620 + # tritonclient
621 + # vocos
622 + # xarray
623 +omegaconf==2.3.0
624 + # via
625 + # hydra-core
626 + # lightning
627 +open-clip-torch==2.32.0
628 + # via -r requirements/rocm-test.in
629 +openai-harmony==0.0.8
630 + # via
631 + # -c requirements/common.txt
632 + # gpt-oss
633 +opencensus==0.11.4
634 + # via ray
635 +opencensus-context==0.1.3
636 + # via opencensus
637 +opencv-python-headless==4.13.0.92
638 + # via
639 + # -c requirements/common.txt
640 + # -r requirements/rocm-test.in
641 + # albumentations
642 + # mistral-common
643 +openpyxl==3.1.5
644 + # via -r requirements/rocm-test.in
645 +opentelemetry-api==1.40.0
646 + # via
647 + # -c requirements/common.txt
648 + # opentelemetry-exporter-prometheus
649 + # opentelemetry-sdk
650 + # opentelemetry-semantic-conventions
651 +opentelemetry-exporter-prometheus==0.61b0
652 + # via ray
653 +opentelemetry-proto==1.40.0
654 + # via ray
655 +opentelemetry-sdk==1.40.0
656 + # via
657 + # -c requirements/common.txt
658 + # opentelemetry-exporter-prometheus
659 + # ray
660 +opentelemetry-semantic-conventions==0.61b0
661 + # via opentelemetry-sdk
662 +optuna==3.6.1
663 + # via genai-perf
664 +orjson==3.11.7
665 + # via
666 + # genai-perf
667 + # kaleido
668 +packaging==26.0
669 + # via
670 + # -c requirements/rocm.txt
671 + # accelerate
672 + # bitsandbytes
673 + # black
674 + # datasets
675 + # evaluate
676 + # fastparquet
677 + # geopandas
678 + # huggingface-hub
679 + # hydra-core
680 + # kaleido
681 + # kornia
682 + # lazy-loader
683 + # lightning
684 + # lightning-utilities
685 + # matplotlib
686 + # optuna
687 + # peft
688 + # plotly
689 + # pooch
690 + # pyogrio
691 + # pytest
692 + # pytest-rerunfailures
693 + # pytorch-lightning
694 + # ray
695 + # rioxarray
696 + # scikit-image
697 + # statsmodels
698 + # tensorboard
699 + # tensorboardx
700 + # torchmetrics
701 + # transformers
702 + # typepy
703 + # wandb
704 + # xarray
705 +pandas==3.0.1
706 + # via
707 + # datasets
708 + # evaluate
709 + # fastparquet
710 + # genai-perf
711 + # geopandas
712 + # statsmodels
713 + # tacoreader
714 + # torchgeo
715 + # xarray
716 +pathspec==1.0.4
717 + # via black
718 +pathvalidate==3.3.1
719 + # via pytablewriter
720 +patsy==1.0.2
721 + # via statsmodels
722 +peft==0.18.1
723 + # via -r requirements/rocm-test.in
724 +perceptron==0.1.4
725 + # via -r requirements/rocm-test.in
726 +perf-analyzer==0.1.0
727 + # via genai-perf
728 +pillow==12.1.1
729 + # via
730 + # diffusers
731 + # genai-perf
732 + # imagehash
733 + # imageio
734 + # lightly-utils
735 + # matplotlib
736 + # mistral-common
737 + # perceptron
738 + # scikit-image
739 + # segmentation-models-pytorch
740 + # tensorboard
741 + # torchgeo
742 + # torchvision
743 platformdirs==4.3.6
32 - # via pooch
744 + # via
745 + # black
746 + # pooch
747 + # python-discovery
748 + # virtualenv
749 + # wandb
750 +plotly==6.6.0
751 + # via
752 + # -r requirements/rocm-test.in
753 + # genai-perf
754 +pluggy==1.6.0
755 + # via
756 + # pytest
757 + # pytest-cov
758 +polars==1.39.3
759 + # via mteb
760 +polars-runtime-32==1.39.3
761 + # via polars
762 pooch==1.8.2
763 # via librosa
35 -soundfile==0.13.1
36 - # via librosa
37 -soxr==0.5.0.post1
38 - # via librosa
39 -librosa==0.10.2.post1
40 -
41 -# Retrieval and search
42 -bm25s==0.2.13
43 - # via mteb
764 +portalocker==3.2.0
765 + # via sacrebleu
766 +pqdm==0.2.0
767 + # via -r requirements/rocm-test.in
768 +prometheus-client==0.24.1
769 + # via
770 + # -c requirements/common.txt
771 + # opentelemetry-exporter-prometheus
772 + # ray
773 +propcache==0.4.1
774 + # via
775 + # aiohttp
776 + # yarl
777 +proto-plus==1.27.1
778 + # via google-api-core
779 +protobuf==6.33.6
780 + # via
781 + # -c requirements/common.txt
782 + # google-api-core
783 + # googleapis-common-protos
784 + # grpcio-reflection
785 + # opentelemetry-proto
786 + # proto-plus
787 + # ray
788 + # tensorboard
789 + # tensorboardx
790 + # tensorizer
791 + # wandb
792 +psutil==7.2.2
793 + # via
794 + # accelerate
795 + # peft
796 + # tensorizer
797 +py==1.11.0
798 + # via pytest-forked
799 +py-spy==0.4.1
800 + # via ray
801 +pyarrow==23.0.1
802 + # via
803 + # datasets
804 + # genai-perf
805 + # tacoreader
806 + # terratorch
807 +pyasn1==0.6.3
808 + # via pyasn1-modules
809 +pyasn1-modules==0.4.2
810 + # via google-auth
811 +pycocotools==2.0.11
812 + # via terratorch
813 +pycountry==26.2.16
814 + # via pydantic-extra-types
815 +pycparser==3.0
816 + # via cffi
817 +pycryptodomex==3.23.0
818 + # via blobfile
819 +pydantic==2.12.5
820 + # via
821 + # -c requirements/common.txt
822 + # -r requirements/rocm-test.in
823 + # albumentations
824 + # datamodel-code-generator
825 + # fastapi
826 + # gpt-oss
827 + # lightly
828 + # mistral-common
829 + # mteb
830 + # openai-harmony
831 + # pydantic-extra-types
832 + # ray
833 + # wandb
834 +pydantic-core==2.41.5
835 + # via pydantic
836 +pydantic-extra-types==2.11.1
837 + # via mistral-common
838 +pygments==2.19.2
839 + # via rich
840 +pyjwt==2.12.1
841 + # via msal
842 +pyogrio==0.12.1
843 + # via geopandas
844 +pyparsing==3.3.2
845 + # via
846 + # matplotlib
847 + # rasterio
848 +pyproj==3.7.2
849 + # via
850 + # geopandas
851 + # rioxarray
852 + # torchgeo
853 +pyrate-limiter==3.9.0
854 + # via schemathesis
855 pystemmer==3.0.0
856 # via mteb
46 -
47 -# Multi-modal processing
48 -av==16.1.0
49 - # required for audio_in_video tests
50 -resampy==0.4.3
51 - # audio processing, required for audio_in_video tests
52 -blobfile==3.0.0
53 - # Multi-Modal Models Test
54 -decord==0.6.0
55 - # video processing, required by entrypoints/openai/chat_completion/test_video.py
857 +pytablewriter==1.2.1
858 + # via lm-eval
859 +pytest==8.3.5
860 + # via
861 + # -r requirements/rocm-test.in
862 + # buildkite-test-collector
863 + # genai-perf
864 + # pytest-asyncio
865 + # pytest-cov
866 + # pytest-forked
867 + # pytest-mock
868 + # pytest-rerunfailures
869 + # pytest-shard
870 + # pytest-subtests
871 + # pytest-timeout
872 + # schemathesis
873 +pytest-asyncio==0.24.0
874 + # via -r requirements/rocm-test.in
875 +pytest-cov==6.3.0
876 + # via -r requirements/rocm-test.in
877 +pytest-forked==1.6.0
878 + # via -r requirements/rocm-test.in
879 +pytest-mock==3.15.1
880 + # via genai-perf
881 +pytest-rerunfailures==14.0
882 + # via -r requirements/rocm-test.in
883 +pytest-shard==0.1.2
884 + # via -r requirements/rocm-test.in
885 +pytest-subtests==0.14.2
886 + # via schemathesis
887 +pytest-timeout==2.3.1
888 + # via -r requirements/rocm-test.in
889 +python-box==7.4.1
890 + # via terratorch
891 +python-dateutil==2.9.0.post0
892 + # via
893 + # arrow
894 + # botocore
895 + # lightly
896 + # matplotlib
897 + # pandas
898 + # typepy
899 +python-discovery==1.2.0
900 + # via virtualenv
901 +python-rapidjson==1.23
902 + # via tritonclient
903 +pytokens==0.4.1
904 + # via black
905 +pytorch-lightning==2.6.1
906 + # via
907 + # lightly
908 + # lightning
909 +pytrec-eval-terrier==0.5.10
910 + # via mteb
911 +pytz==2026.1.post1
912 + # via typepy
913 +pywavelets==1.9.0
914 + # via imagehash
915 +pyyaml==6.0.3
916 + # via
917 + # accelerate
918 + # albumentations
919 + # datamodel-code-generator
920 + # datasets
921 + # genai-perf
922 + # huggingface-hub
923 + # jsonargparse
924 + # lightning
925 + # omegaconf
926 + # optuna
927 + # peft
928 + # pytorch-lightning
929 + # ray
930 + # responses
931 + # schemathesis
932 + # timm
933 + # transformers
934 + # vocos
935 + # wandb
936 rapidfuzz==3.12.1
57 -
58 -# OpenAI compatibility and testing
59 -gpt-oss==0.0.8
60 - # OpenAI compatibility tests
937 + # via
938 + # -r requirements/rocm-test.in
939 + # jiwer
940 +rasterio==1.5.0
941 + # via
942 + # rioxarray
943 + # terratorch
944 + # torchgeo
945 +ray==2.54.0
946 + # via -r requirements/rocm-test.in
947 +redis==7.3.0
948 + # via tensorizer
949 +referencing==0.37.0
950 + # via
951 + # jsonschema
952 + # jsonschema-specifications
953 +regex==2026.2.28
954 + # via
955 + # diffusers
956 + # nltk
957 + # open-clip-torch
958 + # sacrebleu
959 + # tiktoken
960 + # transformers
961 +requests==2.32.5
962 + # via
963 + # -c requirements/common.txt
964 + # azure-core
965 + # buildkite-test-collector
966 + # datasets
967 + # diffusers
968 + # docker
969 + # evaluate
970 + # google-api-core
971 + # google-cloud-storage
972 + # gpt-oss
973 + # huggingface-hub
974 + # lightly
975 + # lm-eval
976 + # mistral-common
977 + # msal
978 + # mteb
979 + # pooch
980 + # ray
981 + # responses
982 + # schemathesis
983 + # starlette-testclient
984 + # tacoreader
985 + # tiktoken
986 + # transformers
987 + # wandb
988 +resampy==0.4.3
989 + # via -r requirements/rocm-test.in
990 +responses==0.26.0
991 + # via genai-perf
992 +rfc3339-validator==0.1.4
993 + # via jsonschema
994 +rfc3987==1.3.8
995 + # via jsonschema
996 +rich==14.3.3
997 + # via
998 + # genai-perf
999 + # lightning
1000 + # mteb
1001 + # perceptron
1002 + # terratorch
1003 + # typer
1004 +rioxarray==0.22.0
1005 + # via terratorch
1006 +rouge-score==0.1.2
1007 + # via lm-eval
1008 +rpds-py==0.30.0
1009 + # via
1010 + # jsonschema
1011 + # referencing
1012 +rtree==1.4.1
1013 + # via torchgeo
1014 +runai-model-streamer==0.15.7
1015 + # via
1016 + # -c requirements/rocm.txt
1017 + # -r requirements/rocm-test.in
1018 +runai-model-streamer-azure==0.15.7
1019 + # via runai-model-streamer
1020 +runai-model-streamer-gcs==0.15.7
1021 + # via runai-model-streamer
1022 +runai-model-streamer-s3==0.15.7
1023 + # via runai-model-streamer
1024 +s3transfer==0.16.0
1025 + # via boto3
1026 +sacrebleu==2.6.0
1027 + # via lm-eval
1028 +safetensors==0.7.0
1029 + # via
1030 + # accelerate
1031 + # diffusers
1032 + # open-clip-torch
1033 + # peft
1034 + # segmentation-models-pytorch
1035 + # timm
1036 + # transformers
1037 schemathesis==3.39.15
62 - # OpenAI schema test
63 -
64 -# Evaluation and benchmarking
65 -lm-eval[api]==0.4.11
66 -jiwer==4.0.0
67 -
68 -# Required for multiprocessed tests that use spawn method, Datasets and Evaluate Test
69 -multiprocess==0.70.16
70 -
71 -# Required for v1/metrics/test_engine_logger_apis.py
72 -ray[cgraph,default]>=2.48.0
73 -
1038 + # via -r requirements/rocm-test.in
1039 +scikit-image==0.26.0
1040 + # via
1041 + # albumentations
1042 + # terratorch
1043 +scikit-learn==1.8.0
1044 + # via
1045 + # albumentations
1046 + # librosa
1047 + # lm-eval
1048 + # mteb
1049 + # sentence-transformers
1050 + # terratorch
1051 +scipy==1.17.1
1052 + # via
1053 + # albumentations
1054 + # bm25s
1055 + # imagehash
1056 + # librosa
1057 + # mteb
1058 + # pytrec-eval-terrier
1059 + # scikit-image
1060 + # scikit-learn
1061 + # sentence-transformers
1062 + # statsmodels
1063 + # vocos
1064 +segmentation-models-pytorch==0.5.0
1065 + # via
1066 + # -r requirements/rocm-test.in
1067 + # terratorch
1068 + # torchgeo
1069 +sentence-transformers==5.3.0
1070 + # via
1071 + # -r requirements/rocm-test.in
1072 + # mteb
1073 +sentry-sdk==2.55.0
1074 + # via wandb
1075 +setuptools==79.0.1
1076 + # via
1077 + # -c requirements/common.txt
1078 + # -c requirements/rocm.txt
1079 + # pytablewriter
1080 + # tensorboard
1081 + # torch
1082 +shapely==2.1.2
1083 + # via
1084 + # geopandas
1085 + # torchgeo
1086 +shellingham==1.5.4
1087 + # via
1088 + # perceptron
1089 + # typer
1090 +simplejson==3.20.2
1091 + # via choreographer
1092 +six==1.17.0
1093 + # via
1094 + # -c requirements/common.txt
1095 + # junit-xml
1096 + # lightly
1097 + # opencensus
1098 + # python-dateutil
1099 + # rfc3339-validator
1100 + # rouge-score
1101 +smart-open==7.5.1
1102 + # via ray
1103 +smmap==5.0.3
1104 + # via gitdb
1105 +sniffio==1.3.1
1106 + # via
1107 + # anyio
1108 + # httpx
1109 +sortedcontainers==2.4.0
1110 + # via hypothesis
1111 +soundfile==0.13.1
1112 + # via
1113 + # -r requirements/rocm-test.in
1114 + # genai-perf
1115 + # librosa
1116 + # mistral-common
1117 +soxr==0.5.0.post1
1118 + # via
1119 + # librosa
1120 + # mistral-common
1121 +sqlalchemy==2.0.48
1122 + # via
1123 + # alembic
1124 + # optuna
1125 +sqlitedict==2.1.0
1126 + # via lm-eval
1127 +starlette==0.52.1
1128 + # via
1129 + # fastapi
1130 + # schemathesis
1131 + # starlette-testclient
1132 +starlette-testclient==0.4.1
1133 + # via schemathesis
1134 +statsmodels==0.14.6
1135 + # via genai-perf
1136 +stringzilla==4.6.0
1137 + # via albucore
1138 +structlog==25.5.0
1139 + # via gpt-oss
1140 +sympy==1.14.0
1141 + # via
1142 + # einx
1143 + # torch
1144 +tabledata==1.3.4
1145 + # via pytablewriter
1146 +tabulate==0.10.0
1147 + # via sacrebleu
1148 +tacoreader==0.5.6
1149 + # via terratorch
1150 +tblib==3.1.0
1151 + # via -r requirements/rocm-test.in
1152 +tcolorpy==0.1.7
1153 + # via pytablewriter
1154 +tenacity==9.1.4
1155 + # via
1156 + # gpt-oss
1157 + # lm-eval
1158 +tensorboard==2.20.0
1159 + # via terratorch
1160 +tensorboard-data-server==0.7.2
1161 + # via tensorboard
1162 +tensorboardx==2.6.4
1163 + # via lightning
1164 +tensorizer==2.10.1
1165 + # via
1166 + # -c requirements/rocm.txt
1167 + # -r requirements/rocm-test.in
1168 +termcolor==3.3.0
1169 + # via
1170 + # gpt-oss
1171 + # terratorch
1172 +terratorch==1.2.2
1173 + # via -r requirements/rocm-test.in
1174 +threadpoolctl==3.6.0
1175 + # via scikit-learn
1176 +tifffile==2026.3.3
1177 + # via
1178 + # scikit-image
1179 + # terratorch
1180 +tiktoken==0.12.0
1181 + # via
1182 + # -c requirements/common.txt
1183 + # gpt-oss
1184 + # lm-eval
1185 + # mistral-common
1186 +timm==1.0.17
1187 + # via
1188 + # -c requirements/rocm.txt
1189 + # -r requirements/rocm-test.in
1190 + # open-clip-torch
1191 + # segmentation-models-pytorch
1192 + # terratorch
1193 + # torchgeo
1194 +tokenizers==0.22.0
1195 + # via
1196 + # -c requirements/common.txt
1197 + # -r requirements/rocm-test.in
1198 + # transformers
1199 +tomli==2.4.0
1200 + # via schemathesis
1201 +tomli-w==1.2.0
1202 + # via schemathesis
1203 torchgeo==0.7.0
1204 + # via
1205 + # -r requirements/rocm-test.in
1206 + # terratorch
1207 +torchmetrics==1.9.0
1208 + # via
1209 + # lightning
1210 + # pytorch-lightning
1211 + # terratorch
1212 + # torchgeo
1213 +tqdm==4.67.3
1214 + # via
1215 + # datasets
1216 + # evaluate
1217 + # huggingface-hub
1218 + # lightly
1219 + # lightning
1220 + # lm-eval
1221 + # mteb
1222 + # nltk
1223 + # open-clip-torch
1224 + # optuna
1225 + # peft
1226 + # pqdm
1227 + # pytorch-lightning
1228 + # segmentation-models-pytorch
1229 + # sentence-transformers
1230 + # tacoreader
1231 + # terratorch
1232 + # transformers
1233 +transformers==4.57.5
1234 + # via
1235 + # -c requirements/common.txt
1236 + # -r requirements/rocm-test.in
1237 + # genai-perf
1238 + # peft
1239 + # sentence-transformers
1240 + # transformers-stream-generator
1241 +transformers-stream-generator==0.0.5
1242 + # via -r requirements/rocm-test.in
1243 +tritonclient==2.66.0
1244 + # via -r requirements/rocm-test.in
1245 +typeguard==4.5.1
1246 + # via inflect
1247 +typepy==1.3.4
1248 + # via
1249 + # dataproperty
1250 + # pytablewriter
1251 + # tabledata
1252 +typer==0.24.1
1253 + # via
1254 + # fastsafetensors
1255 + # perceptron
1256 +typeshed-client==2.9.0
1257 + # via jsonargparse
1258 +typing-extensions==4.15.0
1259 + # via
1260 + # -c requirements/common.txt
1261 + # aiosignal
1262 + # albumentations
1263 + # alembic
1264 + # azure-core
1265 + # azure-identity
1266 + # azure-storage-blob
1267 + # chz
1268 + # fastapi
1269 + # grpcio
1270 + # huggingface-hub
1271 + # librosa
1272 + # lightning
1273 + # lightning-utilities
1274 + # lm-eval
1275 + # mistral-common
1276 + # mteb
1277 + # opentelemetry-api
1278 + # opentelemetry-sdk
1279 + # opentelemetry-semantic-conventions
1280 + # pqdm
1281 + # pydantic
1282 + # pydantic-core
1283 + # pydantic-extra-types
1284 + # pytorch-lightning
1285 + # referencing
1286 + # sentence-transformers
1287 + # sqlalchemy
1288 + # starlette
1289 + # torch
1290 + # torchgeo
1291 + # typeguard
1292 + # typeshed-client
1293 + # typing-inspection
1294 + # wandb
1295 +typing-inspection==0.4.2
1296 + # via
1297 + # fastapi
1298 + # pydantic
1299 +tzdata==2025.3
1300 + # via arrow
1301 +uri-template==1.3.0
1302 + # via jsonschema
1303 +urllib3==2.6.3
1304 + # via
1305 + # blobfile
1306 + # botocore
1307 + # docker
1308 + # lightly
1309 + # requests
1310 + # responses
1311 + # sentry-sdk
1312 + # tritonclient
1313 +uvicorn==0.42.0
1314 + # via gpt-oss
1315 +vector-quantize-pytorch==1.28.0
1316 + # via -r requirements/rocm-test.in
1317 +virtualenv==21.2.0
1318 + # via ray
1319 +vocos==0.1.0
1320 + # via -r requirements/rocm-test.in
1321 +wandb==0.25.1
1322 # via terratorch
76 -# MTEB Benchmark Test
77 -mteb[bm25s]>=2, <3
78 -
79 -# Utilities
80 -num2words==0.5.14
1323 +wcwidth==0.6.0
1324 + # via ftfy
1325 +webcolors==25.10.0
1326 + # via jsonschema
1327 +werkzeug==3.1.6
1328 + # via
1329 + # schemathesis
1330 + # tensorboard
1331 +word2number==1.1
1332 # via lm-eval
82 -pqdm==0.2.0
1333 +wrapt==2.1.2
1334 + # via smart-open
1335 +xarray==2026.2.0
1336 + # via rioxarray
1337 +xxhash==3.6.0
1338 + # via
1339 + # datasets
1340 + # evaluate
1341 +yarl==1.23.0
1342 + # via
1343 + # aiohttp
1344 + # schemathesis
1345 +zipp==3.23.0
1346 + # via importlib-metadata
1347 +zstandard==0.25.0
1348 # via lm-eval
1349
85 -# Required for fastsafetensors test
86 -fastsafetensors @ git+https://github.com/foundation-model-stack/fastsafetensors.git@d6f998a03432b2452f8de2bb5cefb5af9795d459
87 -# Required for suffix decoding test
88 -arctic-inference == 0.1.1
89 -# Required for Nemotron test
90 -open-clip-torch==2.32.0
91 -# Required for isaac Multi-Modal generation test
92 -perceptron==0.1.4
93 -# Required for the multi-modal models test
94 -timm==1.0.17
95 -# Required for plugins test
96 -albumentations==1.4.6
97 -# Pin transformers version
98 -transformers==4.57.5
99 -# Pin HF Hub version
100 -huggingface-hub==0.36.2
101 -# Pin Mistral Common
102 -mistral-common[image,audio]==1.10.0
103 -# Required for Prithvi tests
104 -terratorch==1.2.2
105 -# Required for Prithvi tests
106 -segmentation-models-pytorch==0.5.0
107 -# Required for Prithvi tests
108 -imagehash==4.3.2
109 -# Required for bitsandbytes quantization test
110 -bitsandbytes==0.49.2
111 -# Examples (tensorizer) tests
112 -tensorizer==2.10.1
113 -# Multi-modal models test (`allendou/FireRedASR2-LLM-vllm`)
114 -kaldi-native-fbank==1.22.3
115 -# Pinning numpy version
116 -numpy==2.2.6
1350 +# The following packages were excluded from the output:
1351 +# torch
1352 +# torchvision
1353 +# torchaudio
1354 +# triton
1355 +# cuda-bindings
1356 +# cuda-pathfinder
1357 +# cuda-toolkit
1358 +# cupy-cuda12x
1359 +# nvidia-cublas
1360 +# nvidia-cuda-cupti
1361 +# nvidia-cuda-nvrtc
1362 +# nvidia-cuda-runtime
1363 +# nvidia-cudnn-cu13
1364 +# nvidia-cufft
1365 +# nvidia-cufile
1366 +# nvidia-curand
1367 +# nvidia-cusolver
1368 +# nvidia-cusparse
1369 +# nvidia-cusparselt-cu13
1370 +# nvidia-nccl-cu13
1371 +# nvidia-nvjitlink
1372 +# nvidia-nvshmem-cu13
1373 +# nvidia-nvtx