diff --git a/requirements/test/cpu.txt b/requirements/test/cpu.txt index d7cec7766972..08bfc2a8aff0 100644 --- a/requirements/test/cpu.txt +++ b/requirements/test/cpu.txt @@ -886,7 +886,7 @@ pyzmq==27.1.0 # via -r requirements/test/../common.txt rapidfuzz==3.12.1 # via jiwer -ray==2.48.0 +ray==2.56.0 # via -r requirements/test/cuda.in redis==5.2.0 # via tensorizer diff --git a/requirements/test/cuda.in b/requirements/test/cuda.in index d91cba0601ac..5ee67815849e 100644 --- a/requirements/test/cuda.in +++ b/requirements/test/cuda.in @@ -22,7 +22,7 @@ vector_quantize_pytorch # required for minicpmo_26 test vocos # required for minicpmo_26 test peft>=0.19.1 # required for phi-4-mm test pqdm -ray[cgraph,default]>=2.48.0 # Ray Compiled Graph, required by pipeline parallelism tests +ray[cgraph,default]>=2.56.0 # Ray Compiled Graph, required by pipeline parallelism tests sentence-transformers>=5.2.0 # required for embedding tests soundfile # required for audio tests jiwer # required for audio tests diff --git a/requirements/test/cuda.txt b/requirements/test/cuda.txt index c4a9ea1ae2db..936b8e500cdb 100644 --- a/requirements/test/cuda.txt +++ b/requirements/test/cuda.txt @@ -8,12 +8,10 @@ aiohappyeyeballs==2.6.1 # via aiohttp aiohttp==3.13.3 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # aiohttp-cors # datasets # fsspec - # gpt-oss # lm-eval # ray aiohttp-cors==0.8.1 @@ -21,7 +19,7 @@ aiohttp-cors==0.8.1 aiosignal==1.4.0 # via aiohttp albumentations==1.4.6 - # via -r requirements/test/cuda.in + # via -r cuda.in alembic==1.16.4 # via optuna annotated-doc==0.0.4 @@ -31,9 +29,7 @@ annotated-doc==0.0.4 annotated-types==0.7.0 # via pydantic anthropic==0.112.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt anyio==4.14.1 # via # anthropic @@ -44,11 +40,9 @@ anyio==4.14.1 # starlette # watchfiles apache-tvm-ffi==0.1.10 - # via - # -c requirements/cuda.txt - # xgrammar -arctic-inference==0.1.1 - # via -r requirements/test/cuda.in + # via xgrammar +arctic-inference==0.1.1 ; platform_machine == "x86_64" + # via -r cuda.in argcomplete==3.5.1 # via datamodel-code-generator astor==0.8.1 @@ -62,7 +56,7 @@ attrs==24.2.0 audioread==3.0.1 # via librosa av==16.1.0 - # via -r requirements/test/cuda.in + # via -r cuda.in azure-core==1.38.2 # via # azure-identity @@ -72,15 +66,15 @@ azure-identity==1.25.2 azure-storage-blob==12.28.0 # via runai-model-streamer-azure backoff==2.2.1 - # via -r requirements/test/cuda.in + # via -r cuda.in bitsandbytes==0.49.2 - # via -r requirements/test/cuda.in + # via -r cuda.in black==24.10.0 # via datamodel-code-generator blake3==1.0.9 - # via -r requirements/test/../common.txt + # via -r ../common.txt blobfile==3.0.0 - # via -r requirements/test/cuda.in + # via -r cuda.in bm25s==0.2.13 # via mteb boto3==1.35.57 @@ -94,13 +88,13 @@ botocore==1.35.57 bounded-pool-executor==0.0.3 # via pqdm buildkite-test-collector==0.1.9 - # via -r requirements/test/cuda.in + # via -r cuda.in cachetools==5.5.2 # via - # -r requirements/test/../common.txt + # -r ../common.txt # google-auth cbor2==6.1.2 - # via -r requirements/test/../common.txt + # via -r ../common.txt certifi==2024.8.30 # via # httpcore @@ -115,8 +109,6 @@ chardet==5.2.0 # via mbstrdecoder charset-normalizer==3.4.0 # via requests -chz==0.3.0 - # via gpt-oss click==8.4.2 # via # black @@ -128,9 +120,9 @@ click==8.4.2 # schemathesis # uvicorn cloudpickle==3.1.2 - # via -r requirements/test/../common.txt + # via -r ../common.txt cohere-melody==0.9.0 - # via -r requirements/test/cuda.in + # via -r cuda.in colorama==0.4.6 # via # perceptron @@ -140,12 +132,10 @@ colorful==0.5.6 colorlog==6.10.1 # via optuna compressed-tensors==0.17.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt contourpy==1.3.0 # via matplotlib -coverage==7.10.6 +coverage[toml]==7.10.6 # via pytest-cov cramjam==2.9.0 # via fastparquet @@ -159,32 +149,30 @@ cuda-bindings==13.0.3 # via torch cuda-pathfinder==1.3.3 # via cuda-bindings -cuda-toolkit==13.0.2 +cuda-toolkit[cublas,cudart,cufft,cufile,cupti,curand,cusolver,cusparse,nvjitlink,nvrtc,nvtx]==13.0.2 # via torch cupy-cuda12x==13.6.0 # via ray cycler==0.12.1 # via matplotlib datamodel-code-generator==0.26.3 - # via -r requirements/test/cuda.in + # via -r cuda.in dataproperty==1.0.1 # via # pytablewriter # tabledata datasets==3.3.0 # via - # -r requirements/test/cuda.in + # -r cuda.in # evaluate # lm-eval # mteb decorator==5.1.1 # via librosa -decord==0.6.0 - # via -r requirements/test/cuda.in +decord==0.6.0 ; platform_machine == "x86_64" + # via -r cuda.in depyf==0.20.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt detect-installer==0.1.0 # via fastapi-cloud-cli dill==0.3.8 @@ -202,15 +190,13 @@ distro==1.9.0 # openai dnspython==2.7.0 # via email-validator -docker==7.1.0 - # via gpt-oss docopt==0.6.2 # via num2words docstring-parser==0.18.0 # via anthropic einops==0.8.1 # via - # -r requirements/test/../common.txt + # -r ../common.txt # encodec # vector-quantize-pytorch # vocos @@ -226,13 +212,11 @@ et-xmlfile==2.0.0 # via openpyxl evaluate==0.4.3 # via lm-eval -fastapi==0.136.3 +fastapi[standard]==0.136.3 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # gpt-oss + # -r ../common.txt # model-hosting-container-standards -fastapi-cli==0.0.27 +fastapi-cli[standard]==0.0.27 # via fastapi fastapi-cloud-cli==0.21.0 # via fastapi-cli @@ -245,13 +229,10 @@ fastparquet==2024.11.0 fastrlock==0.8.2 # via cupy-cuda12x fastsafetensors==0.3.2 - # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # via -r cuda.in filelock==3.16.1 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # blobfile # datasets # huggingface-hub @@ -266,7 +247,7 @@ frozenlist==1.5.0 # via # aiohttp # aiosignal -fsspec==2024.12.0 +fsspec[http]==2024.12.0 # via # datasets # evaluate @@ -276,7 +257,7 @@ fsspec==2024.12.0 ftfy==6.3.1 # via open-clip-torch genai-perf==0.0.16 - # via -r requirements/test/cuda.in + # via -r cuda.in genson==1.3.0 # via datamodel-code-generator google-api-core==2.24.2 @@ -305,20 +286,18 @@ googleapis-common-protos==1.70.0 # google-api-core # opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-http -gpt-oss==0.0.8 - # via -r requirements/test/cuda.in graphql-core==3.2.6 # via hypothesis-graphql greenlet==3.2.3 # via sqlalchemy grpcio==1.78.0 # via - # -r requirements/test/cuda.in + # -r cuda.in # grpcio-reflection # opentelemetry-exporter-otlp-proto-grpc # ray grpcio-reflection==1.78.0 - # via -r requirements/test/cuda.in + # via -r cuda.in h11==0.14.0 # via # httpcore @@ -333,15 +312,13 @@ hiredis==3.0.0 # via tensorizer hpack==4.1.0 # via h2 -html2text==2025.4.15 - # via gpt-oss httpcore==1.0.6 # via httpx httptools==0.8.0 # via uvicorn -httpx==0.27.2 +httpx[http2]==0.27.2 # via - # -r requirements/test/cuda.in + # -r cuda.in # anthropic # fastapi # fastapi-cloud-cli @@ -387,9 +364,9 @@ idna==3.10 # requests # yarl ijson==3.5.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt imagehash==4.3.2 - # via -r requirements/test/cuda.in + # via -r cuda.in imageio==2.37.0 # via scikit-image importlib-metadata==8.7.0 @@ -398,8 +375,8 @@ inflect==5.6.2 # via datamodel-code-generator iniconfig==2.0.0 # via pytest -instanttensor==0.1.9 - # via -r requirements/test/cuda.in +instanttensor==0.1.9 ; platform_machine == "x86_64" + # via -r cuda.in interegular==0.3.3 # via lm-format-enforcer isodate==0.7.2 @@ -418,7 +395,7 @@ jiter==0.15.0 # anthropic # openai jiwer==3.0.5 - # via -r requirements/test/cuda.in + # via -r cuda.in jmespath==1.0.1 # via # boto3 @@ -431,8 +408,7 @@ joblib==1.4.2 # scikit-learn jsonschema==4.23.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # hypothesis-jsonschema # mcp # mistral-common @@ -444,15 +420,13 @@ jsonschema-specifications==2024.10.1 junit-xml==1.9 # via schemathesis kaldi-native-fbank==1.22.3 - # via -r requirements/test/cuda.in + # via -r cuda.in kaleido==0.2.1 # via genai-perf kiwisolver==1.4.7 # via matplotlib lark==1.2.2 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt lazy-loader==0.4 # via # librosa @@ -460,25 +434,20 @@ lazy-loader==0.4 libnacl==2.1.0 # via tensorizer librosa==0.10.2.post1 - # via -r requirements/test/cuda.in -llguidance==1.7.6 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r cuda.in +llguidance==1.7.6 ; platform_machine == "x86_64" or platform_machine == "arm64" or platform_machine == "aarch64" or platform_machine == "ppc64le" or platform_machine == "s390x" + # via -r ../common.txt llvmlite==0.47.0 # via numba -lm-eval==0.4.12 - # via -r requirements/test/cuda.in +lm-eval[api]==0.4.12 + # via -r cuda.in lm-format-enforcer==0.11.3 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt loguru==0.7.3 # via compressed-tensors lxml==5.3.0 # via # blobfile - # gpt-oss # sacrebleu mako==1.3.10 # via alembic @@ -490,25 +459,23 @@ markupsafe==3.0.1 # mako # werkzeug matplotlib==3.9.2 - # via -r requirements/test/cuda.in + # via -r cuda.in mbstrdecoder==1.1.3 # via # dataproperty # pytablewriter # typepy mcp==1.28.1 - # via -r requirements/test/../common.txt + # via -r ../common.txt mdurl==0.1.2 # via markdown-it-py -mistral-common==1.11.5 +mistral-common[audio,image,opencv,soundfile,soxr]==1.11.5 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # -r requirements/test/cuda.in + # -r ../common.txt + # -r cuda.in + # mistral-common model-hosting-container-standards==0.1.16 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt more-itertools==10.5.0 # via lm-eval mpmath==1.3.0 @@ -524,9 +491,9 @@ msgpack==1.1.0 # librosa # ray msgspec==0.21.1 - # via -r requirements/test/../common.txt -mteb==2.8.3 - # via -r requirements/test/cuda.in + # via -r ../common.txt +mteb[bm25s]==2.8.3 + # via -r cuda.in multidict==6.1.0 # via # aiohttp @@ -542,19 +509,18 @@ networkx==3.2.1 # scikit-image # torch ninja==1.13.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt nltk==3.9.1 # via rouge-score num2words==0.5.14 - # via -r requirements/test/cuda.in + # via -r cuda.in numba==0.65.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r cuda.in # librosa numpy==2.2.6 # via - # -r requirements/test/../common.txt + # -r ../common.txt # accelerate # albumentations # bitsandbytes @@ -639,41 +605,32 @@ nvidia-nvshmem-cu13==3.4.5 nvidia-nvtx==13.0.85 # via cuda-toolkit open-clip-torch==2.32.0 - # via -r requirements/test/cuda.in + # via -r cuda.in openai==2.44.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt openai-harmony==0.0.4 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # gpt-oss + # via -r ../common.txt opencensus==0.11.4 # via ray opencensus-context==0.1.3 # via opencensus opencv-python-headless==4.13.0.90 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # albumentations # mistral-common openpyxl==3.1.5 - # via -r requirements/test/cuda.in + # via -r cuda.in opentelemetry-api==1.35.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-prometheus # opentelemetry-sdk # opentelemetry-semantic-conventions opentelemetry-exporter-otlp==1.35.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt opentelemetry-exporter-otlp-proto-common==1.35.0 # via # opentelemetry-exporter-otlp-proto-grpc @@ -692,8 +649,7 @@ opentelemetry-proto==1.35.0 # ray opentelemetry-sdk==1.35.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-prometheus @@ -701,17 +657,13 @@ opentelemetry-sdk==1.35.0 opentelemetry-semantic-conventions==0.56b0 # via opentelemetry-sdk opentelemetry-semantic-conventions-ai==0.4.13 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt optuna==3.6.1 # via genai-perf orjson==3.11.5 # via genai-perf outlines-core==0.2.14 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt packaging==24.2 # via # accelerate @@ -744,7 +696,7 @@ pandas==2.2.3 # genai-perf # statsmodels partial-json-parser==0.2.1.1.post7 - # via -r requirements/test/../common.txt + # via -r ../common.txt pathspec==0.12.1 # via black pathvalidate==3.2.1 @@ -752,14 +704,14 @@ pathvalidate==3.2.1 patsy==1.0.1 # via statsmodels peft==0.19.1 - # via -r requirements/test/cuda.in + # via -r cuda.in perceptron==0.1.4 - # via -r requirements/test/cuda.in + # via -r cuda.in perf-analyzer==0.1.0 # via genai-perf pillow==10.4.0 # via - # -r requirements/test/../common.txt + # -r ../common.txt # genai-perf # imagehash # imageio @@ -776,7 +728,7 @@ platformdirs==4.3.6 # virtualenv plotly==5.24.1 # via - # -r requirements/test/cuda.in + # -r cuda.in # genai-perf pluggy==1.5.0 # via @@ -789,18 +741,15 @@ pooch==1.8.2 portalocker==2.10.1 # via sacrebleu pqdm==0.2.0 - # via -r requirements/test/cuda.in + # via -r cuda.in prometheus-client==0.22.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # opentelemetry-exporter-prometheus # prometheus-fastapi-instrumentator # ray prometheus-fastapi-instrumentator==8.0.2 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt propcache==0.2.0 # via # aiohttp @@ -809,8 +758,7 @@ proto-plus==1.26.1 # via google-api-core protobuf==6.33.6 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # google-api-core # googleapis-common-protos # grpcio-reflection @@ -820,14 +768,14 @@ protobuf==6.33.6 # tensorizer psutil==6.1.0 # via - # -r requirements/test/../common.txt + # -r ../common.txt # accelerate # peft # tensorizer py==1.11.0 # via pytest-forked py-cpuinfo==9.0.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt py-spy==0.4.0 # via ray pyarrow==23.0.0 @@ -841,24 +789,22 @@ pyasn1==0.6.1 pyasn1-modules==0.4.2 # via google-auth pybase64==1.4.3 - # via -r requirements/test/../common.txt + # via -r ../common.txt pycountry==24.6.1 # via pydantic-extra-types pycparser==2.22 # via cffi pycryptodomex==3.22.0 # via blobfile -pydantic==2.12.0 +pydantic[email]==2.12.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # albumentations # anthropic # compressed-tensors # datamodel-code-generator # fastapi # fastapi-cloud-cli - # gpt-oss # lm-format-enforcer # mcp # mistral-common @@ -872,7 +818,7 @@ pydantic==2.12.0 # xgrammar pydantic-core==2.41.1 # via pydantic -pydantic-extra-types==2.10.5 +pydantic-extra-types[pycountry]==2.10.5 # via # fastapi # mistral-common @@ -884,7 +830,7 @@ pygments==2.18.0 # via # pytest # rich -pyjwt==2.11.0 +pyjwt[crypto]==2.11.0 # via # mcp # msal @@ -898,7 +844,7 @@ pytablewriter==1.2.0 # via lm-eval pytest==9.1.0 # via - # -r requirements/test/cuda.in + # -r cuda.in # buildkite-test-collector # genai-perf # pytest-asyncio @@ -910,19 +856,19 @@ pytest==9.1.0 # pytest-timeout # schemathesis pytest-asyncio==1.4.0 - # via -r requirements/test/cuda.in + # via -r cuda.in pytest-cov==6.3.0 - # via -r requirements/test/cuda.in + # via -r cuda.in pytest-forked==1.6.0 - # via -r requirements/test/cuda.in + # via -r cuda.in pytest-mock==3.14.0 # via genai-perf pytest-rerunfailures==14.0 - # via -r requirements/test/cuda.in + # via -r cuda.in pytest-shard==0.1.2 - # via -r requirements/test/cuda.in + # via -r cuda.in pytest-timeout==2.3.1 - # via -r requirements/test/cuda.in + # via -r cuda.in python-dateutil==2.9.0.post0 # via # botocore @@ -934,7 +880,7 @@ python-dotenv==1.2.2 # pydantic-settings # uvicorn python-json-logger==4.1.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt python-multipart==0.0.32 # via # fastapi @@ -951,7 +897,7 @@ pywavelets==1.9.0 # via imagehash pyyaml==6.0.2 # via - # -r requirements/test/../common.txt + # -r ../common.txt # accelerate # albumentations # datamodel-code-generator @@ -969,13 +915,11 @@ pyyaml==6.0.2 # uvicorn # vocos pyzmq==27.1.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt rapidfuzz==3.12.1 # via jiwer -ray==2.48.0 - # via -r requirements/test/cuda.in +ray[cgraph,default]==2.56.0 + # via -r cuda.in redis==5.2.0 # via tensorizer referencing==0.35.1 @@ -984,7 +928,7 @@ referencing==0.35.1 # jsonschema-specifications regex==2026.2.28 # via - # -r requirements/test/../common.txt + # -r ../common.txt # nltk # open-clip-torch # sacrebleu @@ -992,16 +936,13 @@ regex==2026.2.28 # transformers requests==2.32.3 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # azure-core # buildkite-test-collector # datasets - # docker # evaluate # google-api-core # google-cloud-storage - # gpt-oss # lm-eval # mistral-common # msal @@ -1037,8 +978,8 @@ rpds-py==0.20.1 # referencing rsa==4.9.1 # via google-auth -runai-model-streamer==0.15.7 - # via -r requirements/test/cuda.in +runai-model-streamer[azure,gcs,s3]==0.15.7 + # via -r cuda.in runai-model-streamer-azure==0.15.7 # via runai-model-streamer runai-model-streamer-gcs==0.15.7 @@ -1051,8 +992,7 @@ sacrebleu==2.4.3 # via lm-eval safetensors==0.8.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # accelerate # open-clip-torch # peft @@ -1060,7 +1000,7 @@ safetensors==0.8.0 # timm # transformers schemathesis==4.21.6 - # via -r requirements/test/cuda.in + # via -r cuda.in scikit-image==0.25.2 # via albumentations scikit-learn==1.5.2 @@ -1083,32 +1023,23 @@ scipy==1.13.1 # statsmodels # vocos segmentation-models-pytorch==0.5.0 - # via -r requirements/test/cuda.in + # via -r cuda.in sentence-transformers==5.2.0 # via - # -r requirements/test/cuda.in + # -r cuda.in # mteb sentencepiece==0.2.1 - # via -r requirements/test/../common.txt + # via -r ../common.txt sentry-sdk==2.63.0 # via fastapi-cloud-cli setproctitle==1.3.7 - # via -r requirements/test/../common.txt -setuptools==77.0.3 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # model-hosting-container-standards - # pytablewriter - # torch + # via -r ../common.txt shellingham==1.5.4 # via # perceptron # typer six==1.16.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt # junit-xml # opencensus # python-dateutil @@ -1124,7 +1055,7 @@ sortedcontainers==2.4.0 # via hypothesis soundfile==0.12.1 # via - # -r requirements/test/cuda.in + # -r cuda.in # genai-perf # librosa # mistral-common @@ -1142,8 +1073,7 @@ sse-starlette==3.4.5 # via mcp starlette==1.3.1 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # fastapi # mcp # model-hosting-container-standards @@ -1154,8 +1084,6 @@ starlette-testclient==0.4.1 # via schemathesis statsmodels==0.14.4 # via genai-perf -structlog==25.4.0 - # via gpt-oss supervisor==4.3.0 # via model-hosting-container-standards sympy==1.13.3 @@ -1167,45 +1095,38 @@ tabledata==1.3.3 tabulate==0.9.0 # via sacrebleu tblib==3.1.0 - # via -r requirements/test/cuda.in + # via -r cuda.in tcolorpy==0.1.6 # via pytablewriter tenacity==9.1.2 # via - # gpt-oss # lm-eval # plotly # schemathesis tensorizer==2.10.1 - # via -r requirements/test/cuda.in -termcolor==3.1.0 - # via gpt-oss + # via -r cuda.in threadpoolctl==3.5.0 # via scikit-learn tifffile==2025.3.30 # via scikit-image tiktoken==0.12.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # gpt-oss + # -r ../common.txt # lm-eval # mistral-common timm==1.0.17 # via - # -r requirements/test/cuda.in + # -r cuda.in # open-clip-torch # segmentation-models-pytorch tokenizers==0.22.2 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # -r requirements/test/cuda.in + # -r ../common.txt + # -r cuda.in # transformers -torch==2.11.0+cu130 +torch==2.11.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r cuda.in # accelerate # bitsandbytes # compressed-tensors @@ -1223,26 +1144,22 @@ torch==2.11.0+cu130 # vector-quantize-pytorch # vocos # xgrammar -torchaudio==2.11.0+cu130 +torchaudio==2.11.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r cuda.in # encodec # vocos -torchcodec==0.14.0+cu130 - # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in -torchvision==0.26.0+cu130 +torchcodec==0.15.0 + # via -r cuda.in +torchvision==0.26.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r cuda.in # open-clip-torch # segmentation-models-pytorch # timm tqdm==4.67.3 # via - # -r requirements/test/../common.txt + # -r ../common.txt # datasets # evaluate # huggingface-hub @@ -1259,9 +1176,8 @@ tqdm==4.67.3 # transformers transformers==5.13.1 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # -r requirements/test/cuda.in + # -r ../common.txt + # -r cuda.in # compressed-tensors # genai-perf # peft @@ -1269,14 +1185,14 @@ transformers==5.13.1 # transformers-stream-generator # xgrammar transformers-stream-generator==0.0.5 - # via -r requirements/test/cuda.in + # via -r cuda.in triton==3.6.0 # via # torch # xgrammar tritonclient==2.64.0 - # via -r requirements/test/cuda.in -typepy==1.3.2 + # via -r cuda.in +typepy[datetime]==1.3.2 # via # dataproperty # pytablewriter @@ -1290,8 +1206,7 @@ typer==0.26.8 # transformers typing-extensions==4.15.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # aiosignal # albumentations # alembic @@ -1301,7 +1216,7 @@ typing-extensions==4.15.0 # azure-core # azure-identity # azure-storage-blob - # chz + # blake3 # fastapi # grpcio # huggingface-hub @@ -1341,29 +1256,27 @@ urllib3==2.2.3 # via # blobfile # botocore - # docker # requests # responses # sentry-sdk # tritonclient -uvicorn==0.35.0 +uvicorn[standard]==0.35.0 # via # fastapi # fastapi-cli # fastapi-cloud-cli - # gpt-oss # mcp uvloop==0.22.1 # via uvicorn vector-quantize-pytorch==1.21.2 - # via -r requirements/test/cuda.in + # via -r cuda.in virtualenv==20.31.2 # via ray vocos==0.1.0 - # via -r requirements/test/cuda.in + # via -r cuda.in watchfiles==1.2.0 # via - # -r requirements/test/../common.txt + # -r ../common.txt # uvicorn wcwidth==0.2.13 # via ftfy @@ -1375,10 +1288,8 @@ word2number==1.1 # via lm-eval wrapt==1.17.2 # via smart-open -xgrammar==0.2.3 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt +xgrammar==0.2.3 ; platform_machine == "x86_64" or platform_machine == "aarch64" or platform_machine == "arm64" or platform_machine == "s390x" or platform_machine == "ppc64le" + # via -r ../common.txt xxhash==3.5.0 # via # datasets diff --git a/requirements/test/nightly-torch.txt b/requirements/test/nightly-torch.txt index b308a4401399..d2e3a5cc073b 100644 --- a/requirements/test/nightly-torch.txt +++ b/requirements/test/nightly-torch.txt @@ -16,7 +16,7 @@ librosa # required for audio tests vocos # required for minicpmo_26 test peft pqdm -ray[cgraph,default]>=2.48.0 # Ray Compiled Graph, required by pipeline parallelism tests +ray[cgraph,default]>=2.56.0 # Ray Compiled Graph, required by pipeline parallelism tests sentence-transformers>=5.2.0 # required for embedding tests soundfile # required for audio tests jiwer # required for audio tests diff --git a/requirements/test/rocm.in b/requirements/test/rocm.in index 7cb3b91f7bd7..d033aa8c586b 100644 --- a/requirements/test/rocm.in +++ b/requirements/test/rocm.in @@ -21,7 +21,7 @@ vector_quantize_pytorch # required for minicpmo_26 test vocos # required for minicpmo_26 test peft>=0.19.1 # required for phi-4-mm test pqdm -ray[cgraph,default]>=2.48.0 # Ray Compiled Graph, required by pipeline parallelism tests +ray[cgraph,default]>=2.56.0 # Ray Compiled Graph, required by pipeline parallelism tests sentence-transformers>=5.2.0 # required for embedding tests soundfile # required for audio tests jiwer # required for audio tests diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index f274a93362ce..75551da40a81 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -8,11 +8,9 @@ aiohappyeyeballs==2.6.1 # via aiohttp aiohttp==3.13.3 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # aiohttp-cors # fsspec - # gpt-oss # lm-eval # ray aiohttp-cors==0.8.1 @@ -20,7 +18,7 @@ aiohttp-cors==0.8.1 aiosignal==1.4.0 # via aiohttp albumentations==1.4.6 - # via -r requirements/test/rocm.in + # via -r rocm.in alembic==1.18.4 # via optuna annotated-doc==0.0.4 @@ -30,9 +28,7 @@ annotated-doc==0.0.4 annotated-types==0.7.0 # via pydantic anthropic==0.93.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt anyio==4.13.0 # via # anthropic @@ -44,11 +40,10 @@ anyio==4.13.0 # watchfiles apache-tvm-ffi==0.1.10 # via - # -c requirements/rocm.txt # tilelang # xgrammar arctic-inference==0.1.1 - # via -r requirements/test/rocm.in + # via -r rocm.in argcomplete==3.6.3 # via datamodel-code-generator astor==0.8.1 @@ -61,7 +56,7 @@ attrs==26.1.0 audioread==3.0.1 # via librosa av==16.1.0 - # via -r requirements/test/rocm.in + # via -r rocm.in azure-core==1.39.0 # via # azure-identity @@ -71,15 +66,15 @@ azure-identity==1.25.3 azure-storage-blob==12.28.0 # via runai-model-streamer-azure backoff==2.2.1 - # via -r requirements/test/rocm.in + # via -r rocm.in bitsandbytes==0.49.2 - # via -r requirements/test/rocm.in + # via -r rocm.in black==26.3.1 # via datamodel-code-generator blake3==1.0.8 - # via -r requirements/test/../common.txt + # via -r ../common.txt blobfile==3.0.0 - # via -r requirements/test/rocm.in + # via -r rocm.in bm25s==0.2.13 # via mteb boto3==1.42.74 @@ -93,11 +88,11 @@ botocore==1.42.74 bounded-pool-executor==0.0.3 # via pqdm buildkite-test-collector==0.1.9 - # via -r requirements/test/rocm.in + # via -r rocm.in cachetools==7.0.5 - # via -r requirements/test/../common.txt + # via -r ../common.txt cbor2==5.9.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt certifi==2026.2.25 # via # httpcore @@ -114,8 +109,6 @@ charset-normalizer==3.4.6 # via requests choreographer==1.2.1 # via kaleido -chz==0.4.0 - # via gpt-oss click==8.4.2 # via # black @@ -129,10 +122,10 @@ click==8.4.2 # uvicorn cloudpickle==3.1.2 # via - # -r requirements/test/../common.txt + # -r ../common.txt # tilelang cohere-melody==0.9.0 - # via -r requirements/test/rocm.in + # via -r rocm.in colorama==0.4.6 # via # perceptron @@ -142,12 +135,10 @@ colorful==0.5.8 colorlog==6.10.1 # via optuna compressed-tensors==0.17.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt contourpy==1.3.3 # via matplotlib -coverage==7.13.5 +coverage[toml]==7.13.5 # via pytest-cov cramjam==2.11.0 # via fastparquet @@ -158,28 +149,36 @@ cryptography==46.0.0 # google-auth # msal # pyjwt +cuda-bindings==13.3.1 + # via torch +cuda-pathfinder==1.6.0 + # via + # cuda-bindings + # cupy-cuda12x +cuda-toolkit[cublas,cudart,cufft,cufile,cupti,curand,cusolver,cusparse,nvjitlink,nvrtc,nvtx]==13.0.3.0 + # via torch +cupy-cuda12x==14.1.1 + # via ray cycler==0.12.1 # via matplotlib datamodel-code-generator==0.55.0 - # via -r requirements/test/rocm.in + # via -r rocm.in dataproperty==1.1.0 # via # pytablewriter # tabledata datasets==3.6.0 # via - # -r requirements/test/rocm.in + # -r rocm.in # evaluate # lm-eval # mteb decorator==5.2.1 # via librosa decord==0.6.0 - # via -r requirements/test/rocm.in + # via -r rocm.in depyf==0.20.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt dill==0.3.8 # via # datasets @@ -195,15 +194,13 @@ distro==1.9.0 # openai dnspython==2.8.0 # via email-validator -docker==7.1.0 - # via gpt-oss docopt==0.6.2 # via num2words docstring-parser==0.17.0 # via anthropic einops==0.8.2 # via - # -r requirements/test/../common.txt + # -r ../common.txt # encodec # vector-quantize-pytorch # vocos @@ -219,13 +216,11 @@ et-xmlfile==2.0.0 # via openpyxl evaluate==0.4.6 # via lm-eval -fastapi==0.135.2 +fastapi[standard]==0.135.2 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # gpt-oss + # -r ../common.txt # model-hosting-container-standards -fastapi-cli==0.0.24 +fastapi-cli[standard]==0.0.24 # via fastapi fastapi-cloud-cli==0.16.1 # via fastapi-cli @@ -234,13 +229,10 @@ fastar==0.10.0 fastparquet==2026.3.0 # via genai-perf fastsafetensors==0.3.2 - # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in + # via -r rocm.in filelock==3.25.2 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # blobfile # datasets # huggingface-hub @@ -256,7 +248,7 @@ frozenlist==1.8.0 # via # aiohttp # aiosignal -fsspec==2025.3.0 +fsspec[http]==2025.3.0 # via # datasets # evaluate @@ -266,7 +258,7 @@ fsspec==2025.3.0 ftfy==6.3.1 # via open-clip-torch genai-perf==0.0.16 - # via -r requirements/test/rocm.in + # via -r rocm.in genson==1.3.0 # via datamodel-code-generator google-api-core==2.30.0 @@ -295,23 +287,18 @@ googleapis-common-protos==1.73.0 # google-api-core # opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-http -gpt-oss==0.0.8 - # via -r requirements/test/rocm.in graphql-core==3.2.8 # via hypothesis-graphql greenlet==3.3.2 # via sqlalchemy grpcio==1.78.0 # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in + # -r rocm.in # grpcio-reflection # opentelemetry-exporter-otlp-proto-grpc # ray grpcio-reflection==1.78.0 - # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in + # via -r rocm.in h11==0.16.0 # via # httpcore @@ -326,15 +313,13 @@ hiredis==3.3.1 # via tensorizer hpack==4.1.0 # via h2 -html2text==2025.4.15 - # via gpt-oss httpcore==1.0.9 # via httpx httptools==0.7.1 # via uvicorn -httpx==0.27.2 +httpx[http2]==0.27.2 # via - # -r requirements/test/rocm.in + # -r rocm.in # anthropic # fastapi # fastapi-cloud-cli @@ -380,9 +365,9 @@ idna==3.11 # requests # yarl ijson==3.5.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt imagehash==4.3.2 - # via -r requirements/test/rocm.in + # via -r rocm.in imageio==2.37.3 # via scikit-image importlib-metadata==8.7.1 @@ -392,7 +377,7 @@ inflect==7.5.0 iniconfig==2.3.0 # via pytest instanttensor==0.1.9 - # via -r requirements/test/rocm.in + # via -r rocm.in interegular==0.3.3 # via lm-format-enforcer isodate==0.7.2 @@ -411,7 +396,7 @@ jiter==0.14.0 # anthropic # openai jiwer==4.0.0 - # via -r requirements/test/rocm.in + # via -r rocm.in jmespath==1.1.0 # via # boto3 @@ -424,8 +409,7 @@ joblib==1.5.3 # scikit-learn jsonschema==4.26.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # hypothesis-jsonschema # mcp # mistral-common @@ -437,15 +421,13 @@ jsonschema-specifications==2025.9.1 junit-xml==1.9 # via schemathesis kaldi-native-fbank==1.22.3 - # via -r requirements/test/rocm.in + # via -r rocm.in kaleido==1.0.0 # via genai-perf kiwisolver==1.5.0 # via matplotlib lark==1.2.2 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt lazy-loader==0.4 # via # librosa @@ -453,19 +435,15 @@ lazy-loader==0.4 libnacl==2.1.0 # via tensorizer librosa==0.10.2.post1 - # via -r requirements/test/rocm.in -llguidance==1.7.5 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r rocm.in +llguidance==1.7.5 ; platform_machine == "x86_64" or platform_machine == "arm64" or platform_machine == "aarch64" or platform_machine == "ppc64le" or platform_machine == "s390x" + # via -r ../common.txt llvmlite==0.47.0 # via numba -lm-eval==0.4.12 - # via -r requirements/test/rocm.in +lm-eval[api]==0.4.12 + # via -r rocm.in lm-format-enforcer==0.11.3 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt logistro==2.0.1 # via # choreographer @@ -475,7 +453,6 @@ loguru==0.7.3 lxml==6.0.2 # via # blobfile - # gpt-oss # sacrebleu mako==1.3.10 # via alembic @@ -487,27 +464,25 @@ markupsafe==3.0.3 # mako # werkzeug matplotlib==3.10.8 - # via -r requirements/test/rocm.in + # via -r rocm.in mbstrdecoder==1.1.4 # via # dataproperty # pytablewriter # typepy mcp==1.27.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt mdurl==0.1.2 # via markdown-it-py -mistral-common==1.11.5 +mistral-common[audio,image,opencv,soundfile,soxr]==1.11.5 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # -r requirements/test/rocm.in + # -r ../common.txt + # -r rocm.in + # mistral-common ml-dtypes==0.5.4 # via tilelang model-hosting-container-standards==0.1.14 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt more-itertools==10.8.0 # via # inflect @@ -525,16 +500,16 @@ msgpack==1.1.2 # librosa # ray msgspec==0.21.0 - # via -r requirements/test/../common.txt -mteb==2.11.5 - # via -r requirements/test/rocm.in + # via -r ../common.txt +mteb[bm25s]==2.11.5 + # via -r rocm.in multidict==6.7.1 # via # aiohttp # yarl multiprocess==0.70.16 # via - # -r requirements/test/rocm.in + # -r rocm.in # datasets # evaluate mypy-extensions==1.1.0 @@ -546,19 +521,18 @@ networkx==3.6.1 # scikit-image # torch ninja==1.13.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt nltk==3.9.3 # via rouge-score num2words==0.5.14 - # via -r requirements/test/rocm.in + # via -r rocm.in numba==0.65.0 # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in + # -r rocm.in # librosa numpy==2.2.6 # via - # -r requirements/test/../common.txt + # -r ../common.txt # accelerate # albumentations # bitsandbytes @@ -608,42 +582,74 @@ numpy==2.2.6 # tritonclient # vocos # xgrammar +nvidia-cublas==13.1.1.3 + # via + # cuda-toolkit + # nvidia-cudnn-cu13 + # nvidia-cusolver +nvidia-cuda-cupti==13.0.85 + # via cuda-toolkit +nvidia-cuda-nvrtc==13.0.88 + # via + # cuda-toolkit + # nvidia-cublas +nvidia-cuda-runtime==13.0.96 + # via cuda-toolkit +nvidia-cudnn-cu13==9.20.0.48 + # via torch +nvidia-cufft==12.0.0.61 + # via cuda-toolkit +nvidia-cufile==1.15.1.6 + # via cuda-toolkit +nvidia-curand==10.4.0.35 + # via cuda-toolkit +nvidia-cusolver==12.0.4.66 + # via cuda-toolkit +nvidia-cusparse==12.6.3.3 + # via + # cuda-toolkit + # nvidia-cusolver +nvidia-cusparselt-cu13==0.8.1 + # via torch +nvidia-nccl-cu13==2.29.7 + # via torch +nvidia-nvjitlink==13.3.33 + # via + # cuda-toolkit + # nvidia-cufft + # nvidia-cusolver + # nvidia-cusparse +nvidia-nvshmem-cu13==3.4.5 + # via torch +nvidia-nvtx==13.0.85 + # via cuda-toolkit open-clip-torch==2.32.0 - # via -r requirements/test/rocm.in + # via -r rocm.in openai==2.31.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt openai-harmony==0.0.8 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # gpt-oss + # via -r ../common.txt opencensus==0.11.4 # via ray opencensus-context==0.1.3 # via opencensus opencv-python-headless==4.13.0.92 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # albumentations # mistral-common openpyxl==3.1.5 - # via -r requirements/test/rocm.in + # via -r rocm.in opentelemetry-api==1.40.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-prometheus # opentelemetry-sdk # opentelemetry-semantic-conventions opentelemetry-exporter-otlp==1.40.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt opentelemetry-exporter-otlp-proto-common==1.40.0 # via # opentelemetry-exporter-otlp-proto-grpc @@ -662,8 +668,7 @@ opentelemetry-proto==1.40.0 # ray opentelemetry-sdk==1.40.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # opentelemetry-exporter-otlp-proto-grpc # opentelemetry-exporter-otlp-proto-http # opentelemetry-exporter-prometheus @@ -674,9 +679,7 @@ opentelemetry-semantic-conventions==0.61b0 # opentelemetry-sdk # opentelemetry-semantic-conventions-ai opentelemetry-semantic-conventions-ai==0.5.1 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt optuna==3.6.1 # via genai-perf orjson==3.11.7 @@ -684,12 +687,9 @@ orjson==3.11.7 # genai-perf # kaleido outlines-core==0.2.14 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt packaging==26.0 # via - # -c requirements/rocm.txt # accelerate # bitsandbytes # black @@ -720,7 +720,7 @@ pandas==3.0.1 # genai-perf # statsmodels partial-json-parser==0.2.1.1.post7 - # via -r requirements/test/../common.txt + # via -r ../common.txt pathspec==1.0.4 # via black pathvalidate==3.3.1 @@ -728,14 +728,14 @@ pathvalidate==3.3.1 patsy==1.0.2 # via statsmodels peft==0.19.1 - # via -r requirements/test/rocm.in + # via -r rocm.in perceptron==0.1.4 - # via -r requirements/test/rocm.in + # via -r rocm.in perf-analyzer==0.1.0 # via genai-perf pillow==12.1.1 # via - # -r requirements/test/../common.txt + # -r ../common.txt # genai-perf # imagehash # imageio @@ -753,7 +753,7 @@ platformdirs==4.3.6 # virtualenv plotly==6.6.0 # via - # -r requirements/test/rocm.in + # -r rocm.in # genai-perf pluggy==1.6.0 # via @@ -768,18 +768,15 @@ pooch==1.8.2 portalocker==3.2.0 # via sacrebleu pqdm==0.2.0 - # via -r requirements/test/rocm.in + # via -r rocm.in prometheus-client==0.24.1 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # opentelemetry-exporter-prometheus # prometheus-fastapi-instrumentator # ray prometheus-fastapi-instrumentator==8.0.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt propcache==0.4.1 # via # aiohttp @@ -788,8 +785,7 @@ proto-plus==1.27.1 # via google-api-core protobuf==6.33.6 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # google-api-core # googleapis-common-protos # grpcio-reflection @@ -799,7 +795,7 @@ protobuf==6.33.6 # tensorizer psutil==7.2.2 # via - # -r requirements/test/../common.txt + # -r ../common.txt # accelerate # peft # tensorizer @@ -807,7 +803,7 @@ psutil==7.2.2 py==1.11.0 # via pytest-forked py-cpuinfo==9.0.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt py-spy==0.4.1 # via ray pyarrow==23.0.1 @@ -819,24 +815,22 @@ pyasn1==0.6.3 pyasn1-modules==0.4.2 # via google-auth pybase64==1.4.3 - # via -r requirements/test/../common.txt + # via -r ../common.txt pycountry==26.2.16 # via pydantic-extra-types pycparser==3.0 # via cffi pycryptodomex==3.23.0 # via blobfile -pydantic==2.12.5 +pydantic[email]==2.12.5 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # albumentations # anthropic # compressed-tensors # datamodel-code-generator # fastapi # fastapi-cloud-cli - # gpt-oss # lm-format-enforcer # mcp # mistral-common @@ -850,7 +844,7 @@ pydantic==2.12.5 # xgrammar pydantic-core==2.41.5 # via pydantic -pydantic-extra-types==2.11.1 +pydantic-extra-types[pycountry]==2.11.1 # via # fastapi # mistral-common @@ -862,7 +856,7 @@ pygments==2.19.2 # via # pytest # rich -pyjwt==2.12.1 +pyjwt[crypto]==2.12.1 # via # mcp # msal @@ -876,7 +870,7 @@ pytablewriter==1.2.1 # via lm-eval pytest==9.1.0 # via - # -r requirements/test/rocm.in + # -r rocm.in # buildkite-test-collector # genai-perf # pytest-asyncio @@ -888,19 +882,19 @@ pytest==9.1.0 # pytest-timeout # schemathesis pytest-asyncio==1.4.0 - # via -r requirements/test/rocm.in + # via -r rocm.in pytest-cov==6.3.0 - # via -r requirements/test/rocm.in + # via -r rocm.in pytest-forked==1.6.0 - # via -r requirements/test/rocm.in + # via -r rocm.in pytest-mock==3.15.1 # via genai-perf pytest-rerunfailures==14.0 - # via -r requirements/test/rocm.in + # via -r rocm.in pytest-shard==0.1.2 - # via -r requirements/test/rocm.in + # via -r rocm.in pytest-timeout==2.3.1 - # via -r requirements/test/rocm.in + # via -r rocm.in python-dateutil==2.9.0.post0 # via # botocore @@ -914,7 +908,7 @@ python-dotenv==1.2.2 # pydantic-settings # uvicorn python-json-logger==4.1.0 - # via -r requirements/test/../common.txt + # via -r ../common.txt python-multipart==0.0.26 # via # fastapi @@ -931,7 +925,7 @@ pywavelets==1.9.0 # via imagehash pyyaml==6.0.3 # via - # -r requirements/test/../common.txt + # -r ../common.txt # accelerate # albumentations # datamodel-code-generator @@ -949,15 +943,13 @@ pyyaml==6.0.3 # uvicorn # vocos pyzmq==27.1.0 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r ../common.txt rapidfuzz==3.12.1 # via - # -r requirements/test/rocm.in + # -r rocm.in # jiwer -ray==2.54.0 - # via -r requirements/test/rocm.in +ray[cgraph,default]==2.56.0 + # via -r rocm.in redis==7.3.0 # via tensorizer referencing==0.37.0 @@ -966,7 +958,7 @@ referencing==0.37.0 # jsonschema-specifications regex==2026.2.28 # via - # -r requirements/test/../common.txt + # -r ../common.txt # nltk # open-clip-torch # sacrebleu @@ -974,16 +966,13 @@ regex==2026.2.28 # transformers requests==2.32.5 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # azure-core # buildkite-test-collector # datasets - # docker # evaluate # google-api-core # google-cloud-storage - # gpt-oss # lm-eval # mistral-common # msal @@ -1017,10 +1006,8 @@ rpds-py==0.30.0 # via # jsonschema # referencing -runai-model-streamer==0.15.7 - # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in +runai-model-streamer[azure,gcs,s3]==0.15.7 + # via -r rocm.in runai-model-streamer-azure==0.15.7 # via runai-model-streamer runai-model-streamer-gcs==0.15.7 @@ -1033,8 +1020,7 @@ sacrebleu==2.6.0 # via lm-eval safetensors==0.8.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # accelerate # open-clip-torch # peft @@ -1042,7 +1028,7 @@ safetensors==0.8.0 # timm # transformers schemathesis==4.21.6 - # via -r requirements/test/rocm.in + # via -r rocm.in scikit-image==0.26.0 # via albumentations scikit-learn==1.8.0 @@ -1066,25 +1052,17 @@ scipy==1.17.1 # statsmodels # vocos segmentation-models-pytorch==0.5.0 - # via -r requirements/test/rocm.in + # via -r rocm.in sentence-transformers==5.3.0 # via - # -r requirements/test/rocm.in + # -r rocm.in # mteb sentencepiece==0.2.1 - # via -r requirements/test/../common.txt + # via -r ../common.txt sentry-sdk==2.55.0 # via fastapi-cloud-cli setproctitle==1.3.7 - # via -r requirements/test/../common.txt -setuptools==79.0.1 - # via - # -c requirements/common.txt - # -c requirements/rocm.txt - # -r requirements/test/../common.txt - # model-hosting-container-standards - # pytablewriter - # torch + # via -r ../common.txt shellingham==1.5.4 # via # perceptron @@ -1093,8 +1071,6 @@ simplejson==3.20.2 # via choreographer six==1.17.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt # junit-xml # opencensus # python-dateutil @@ -1110,7 +1086,7 @@ sortedcontainers==2.4.0 # via hypothesis soundfile==0.13.1 # via - # -r requirements/test/rocm.in + # -r rocm.in # genai-perf # librosa # mistral-common @@ -1128,8 +1104,7 @@ sse-starlette==3.3.4 # via mcp starlette==1.3.1 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # fastapi # mcp # model-hosting-container-standards @@ -1140,8 +1115,6 @@ starlette-testclient==0.4.1 # via schemathesis statsmodels==0.14.6 # via genai-perf -structlog==25.5.0 - # via gpt-oss supervisor==4.3.0 # via model-hosting-container-standards sympy==1.14.0 @@ -1153,52 +1126,73 @@ tabledata==1.3.4 tabulate==0.10.0 # via sacrebleu tblib==3.1.0 - # via -r requirements/test/rocm.in + # via -r rocm.in tcolorpy==0.1.7 # via pytablewriter tenacity==9.1.4 # via - # gpt-oss # lm-eval # schemathesis tensorizer==2.10.1 - # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in -termcolor==3.3.0 - # via gpt-oss + # via -r rocm.in threadpoolctl==3.6.0 # via scikit-learn tifffile==2026.3.3 # via scikit-image tiktoken==0.12.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # gpt-oss + # -r ../common.txt # lm-eval # mistral-common tilelang==0.1.10 - # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in + # via -r rocm.in timm==1.0.17 # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in + # -r rocm.in # open-clip-torch # segmentation-models-pytorch tokenizers==0.22.2 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # -r requirements/test/rocm.in + # -r ../common.txt + # -r rocm.in # transformers +tomli==2.4.1 + # via datamodel-code-generator +torch==2.13.0 + # via + # accelerate + # bitsandbytes + # compressed-tensors + # encodec + # instanttensor + # mteb + # open-clip-torch + # peft + # runai-model-streamer + # segmentation-models-pytorch + # sentence-transformers + # tensorizer + # tilelang + # timm + # torch-c-dlpack-ext + # torchvision + # vector-quantize-pytorch + # vocos + # xgrammar torch-c-dlpack-ext==0.1.5 # via tilelang +torchaudio==2.11.0 + # via + # encodec + # vocos +torchvision==0.28.0 + # via + # open-clip-torch + # segmentation-models-pytorch + # timm tqdm==4.67.3 # via - # -r requirements/test/../common.txt + # -r ../common.txt # datasets # evaluate # huggingface-hub @@ -1216,9 +1210,8 @@ tqdm==4.67.3 # transformers transformers==5.13.1 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # -r requirements/test/rocm.in + # -r ../common.txt + # -r rocm.in # compressed-tensors # genai-perf # peft @@ -1226,12 +1219,16 @@ transformers==5.13.1 # transformers-stream-generator # xgrammar transformers-stream-generator==0.0.5 - # via -r requirements/test/rocm.in + # via -r rocm.in +triton==3.7.1 + # via + # torch + # xgrammar tritonclient==2.66.0 - # via -r requirements/test/rocm.in + # via -r rocm.in typeguard==4.5.1 # via inflect -typepy==1.3.4 +typepy[datetime]==1.3.4 # via # dataproperty # pytablewriter @@ -1245,8 +1242,7 @@ typer==0.24.1 # transformers typing-extensions==4.15.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r ../common.txt # aiosignal # albumentations # alembic @@ -1256,7 +1252,7 @@ typing-extensions==4.15.0 # azure-core # azure-identity # azure-storage-blob - # chz + # blake3 # fastapi # grpcio # huggingface-hub @@ -1297,29 +1293,27 @@ urllib3==2.6.3 # via # blobfile # botocore - # docker # requests # responses # sentry-sdk # tritonclient -uvicorn==0.42.0 +uvicorn[standard]==0.42.0 # via # fastapi # fastapi-cli # fastapi-cloud-cli - # gpt-oss # mcp uvloop==0.22.1 # via uvicorn vector-quantize-pytorch==1.28.0 - # via -r requirements/test/rocm.in + # via -r rocm.in virtualenv==21.2.0 # via ray vocos==0.1.0 - # via -r requirements/test/rocm.in + # via -r rocm.in watchfiles==1.1.1 # via - # -r requirements/test/../common.txt + # -r ../common.txt # uvicorn wcwidth==0.6.0 # via ftfy @@ -1331,10 +1325,8 @@ word2number==1.1 # via lm-eval wrapt==2.1.2 # via smart-open -xgrammar==0.2.1 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt +xgrammar==0.2.1 ; platform_machine == "x86_64" or platform_machine == "aarch64" or platform_machine == "arm64" or platform_machine == "s390x" or platform_machine == "ppc64le" + # via -r ../common.txt xxhash==3.6.0 # via # datasets @@ -1345,28 +1337,3 @@ z3-solver==4.15.4.0 # via tilelang zipp==3.23.0 # via importlib-metadata - -# The following packages were excluded from the output: -# torch -# torchvision -# torchaudio -# triton -# cuda-bindings -# cuda-pathfinder -# cuda-toolkit -# cupy-cuda12x -# nvidia-cublas -# nvidia-cuda-cupti -# nvidia-cuda-nvrtc -# nvidia-cuda-runtime -# nvidia-cufft -# nvidia-cufile -# nvidia-curand -# nvidia-cusolver -# nvidia-cusparse -# nvidia-nvjitlink -# nvidia-nvtx -# nvidia-cudnn-cu13 -# nvidia-cusparselt-cu13 -# nvidia-nccl-cu13 -# nvidia-nvshmem-cu13