diff --git a/requirements/test/cuda.txt b/requirements/test/cuda.txt index f504c69c48f4..b40e2dee7662 100644 --- a/requirements/test/cuda.txt +++ b/requirements/test/cuda.txt @@ -8,11 +8,9 @@ aiohappyeyeballs==2.6.1 # via aiohttp aiohttp==3.13.3 # via - # -c requirements/common.txt # aiohttp-cors # datasets # fsspec - # gpt-oss # lm-eval # ray aiohttp-cors==0.8.1 @@ -20,19 +18,17 @@ aiohttp-cors==0.8.1 aiosignal==1.4.0 # via aiohttp albumentations==1.4.6 - # via -r requirements/test/cuda.in + # via -r test/cuda.in alembic==1.16.4 # via optuna -annotated-doc==0.0.4 - # via fastapi annotated-types==0.7.0 # via pydantic anyio==4.6.2.post1 # via # httpx # starlette -arctic-inference==0.1.1 - # via -r requirements/test/cuda.in +arctic-inference==0.1.1 ; platform_machine == "x86_64" + # via -r test/cuda.in argcomplete==3.5.1 # via datamodel-code-generator attrs==24.2.0 @@ -44,7 +40,7 @@ attrs==24.2.0 audioread==3.0.1 # via librosa av==16.1.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in azure-core==1.38.2 # via # azure-identity @@ -54,13 +50,13 @@ 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 test/cuda.in bitsandbytes==0.49.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in black==24.10.0 # via datamodel-code-generator blobfile==3.0.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in bm25s==0.2.13 # via mteb boto3==1.35.57 @@ -74,7 +70,7 @@ 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 test/cuda.in cachetools==5.5.2 # via google-auth certifi==2024.8.30 @@ -90,8 +86,6 @@ chardet==5.2.0 # via mbstrdecoder charset-normalizer==3.4.0 # via requests -chz==0.3.0 - # via gpt-oss click==8.1.7 # via # black @@ -100,9 +94,8 @@ click==8.1.7 # ray # schemathesis # typer - # uvicorn cohere-melody==0.9.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in colorama==0.4.6 # via # perceptron @@ -113,7 +106,7 @@ colorlog==6.10.1 # via optuna contourpy==1.3.0 # via matplotlib -coverage==7.10.6 +coverage[toml]==7.10.6 # via pytest-cov cramjam==2.9.0 # via fastparquet @@ -127,28 +120,28 @@ 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 +datamodel-code-generator==0.64.0 + # via -r test/cuda.in dataproperty==1.0.1 # via # pytablewriter # tabledata datasets==3.3.0 # via - # -r requirements/test/cuda.in + # -r test/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 test/cuda.in dill==0.3.8 # via # datasets @@ -157,43 +150,30 @@ dill==0.3.8 # multiprocess distlib==0.3.9 # via virtualenv -dnspython==2.7.0 - # via email-validator -docker==7.1.0 - # via gpt-oss docopt==0.6.2 # via num2words einops==0.8.1 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # encodec # vector-quantize-pytorch # vocos einx==0.3.0 # via vector-quantize-pytorch -email-validator==2.2.0 - # via pydantic encodec==0.1.1 # via vocos et-xmlfile==2.0.0 # via openpyxl evaluate==0.4.3 # via lm-eval -fastapi==0.136.3 - # via - # -c requirements/common.txt - # gpt-oss fastparquet==2024.11.0 # via genai-perf fastrlock==0.8.2 # via cupy-cuda12x fastsafetensors==0.3.2 - # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # via -r test/cuda.in filelock==3.16.1 # via - # -c requirements/common.txt # blobfile # datasets # huggingface-hub @@ -208,7 +188,7 @@ frozenlist==1.5.0 # via # aiohttp # aiosignal -fsspec==2024.12.0 +fsspec[http]==2024.12.0 # via # datasets # evaluate @@ -218,7 +198,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 test/cuda.in genson==1.3.0 # via datamodel-code-generator google-api-core==2.24.2 @@ -244,23 +224,19 @@ google-resumable-media==2.7.2 # via google-cloud-storage googleapis-common-protos==1.70.0 # via google-api-core -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 test/cuda.in # grpcio-reflection # ray grpcio-reflection==1.78.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in h11==0.14.0 - # via - # httpcore - # uvicorn + # via httpcore h2==4.3.0 # via httpx harfile==0.5.0 @@ -271,13 +247,11 @@ 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 -httpx==0.27.2 +httpx[http2]==0.27.2 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # huggingface-hub # perceptron # schemathesis @@ -310,12 +284,11 @@ hypothesis-jsonschema==0.23.1 idna==3.10 # via # anyio - # email-validator # httpx # requests # yarl imagehash==4.3.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in imageio==2.37.0 # via scikit-image importlib-metadata==8.7.0 @@ -324,8 +297,8 @@ inflect==5.6.2 # via datamodel-code-generator iniconfig==2.0.0 # via pytest -instanttensor==0.1.5 - # via -r requirements/test/cuda.in +instanttensor==0.1.5 ; platform_machine == "x86_64" + # via -r test/cuda.in isodate==0.7.2 # via azure-storage-blob isort==5.13.2 @@ -337,7 +310,7 @@ jinja2==3.1.6 # lm-eval # torch jiwer==3.0.5 - # via -r requirements/test/cuda.in + # via -r test/cuda.in jmespath==1.0.1 # via # boto3 @@ -349,7 +322,6 @@ joblib==1.4.2 # scikit-learn jsonschema==4.23.0 # via - # -c requirements/common.txt # hypothesis-jsonschema # mistral-common # ray @@ -360,7 +332,7 @@ 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 test/cuda.in kaleido==0.2.1 # via genai-perf kiwisolver==1.4.7 @@ -372,15 +344,14 @@ lazy-loader==0.4 libnacl==2.1.0 # via tensorizer librosa==0.10.2.post1 - # via -r requirements/test/cuda.in + # via -r test/cuda.in 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 test/cuda.in lxml==5.3.0 # via # blobfile - # gpt-oss # sacrebleu mako==1.3.10 # via alembic @@ -392,7 +363,7 @@ markupsafe==3.0.1 # mako # werkzeug matplotlib==3.9.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in mbstrdecoder==1.1.3 # via # dataproperty @@ -400,10 +371,10 @@ mbstrdecoder==1.1.3 # typepy 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/cuda.in + # -r test/cuda.in + # mistral-common more-itertools==10.5.0 # via lm-eval mpmath==1.3.0 @@ -418,8 +389,8 @@ msgpack==1.1.0 # via # librosa # ray -mteb==2.8.3 - # via -r requirements/test/cuda.in +mteb[bm25s]==2.8.3 + # via -r test/cuda.in multidict==6.1.0 # via # aiohttp @@ -437,15 +408,14 @@ networkx==3.2.1 nltk==3.9.1 # via rouge-score num2words==0.5.14 - # via -r requirements/test/cuda.in + # via -r test/cuda.in numba==0.65.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # librosa numpy==2.2.6 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # accelerate # albumentations # bitsandbytes @@ -529,26 +499,20 @@ 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 -openai-harmony==0.0.4 - # via - # -c requirements/common.txt - # gpt-oss + # via -r test/cuda.in 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/cuda.in + # -r test/cuda.in # albumentations # mistral-common openpyxl==3.1.5 - # via -r requirements/test/cuda.in + # via -r test/cuda.in opentelemetry-api==1.35.0 # via - # -c requirements/common.txt # opentelemetry-exporter-prometheus # opentelemetry-sdk # opentelemetry-semantic-conventions @@ -558,7 +522,6 @@ opentelemetry-proto==1.35.0 # via ray opentelemetry-sdk==1.35.0 # via - # -c requirements/common.txt # opentelemetry-exporter-prometheus # ray opentelemetry-semantic-conventions==0.56b0 @@ -572,7 +535,6 @@ packaging==24.2 # accelerate # bitsandbytes # black - # datamodel-code-generator # datasets # evaluate # fastparquet @@ -604,9 +566,9 @@ pathvalidate==3.2.1 patsy==1.0.1 # via statsmodels peft==0.18.1 - # via -r requirements/test/cuda.in + # via -r test/cuda.in perceptron==0.1.4 - # via -r requirements/test/cuda.in + # via -r test/cuda.in perf-analyzer==0.1.0 # via genai-perf pillow==10.4.0 @@ -627,7 +589,7 @@ platformdirs==4.3.6 # virtualenv plotly==5.24.1 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # genai-perf pluggy==1.5.0 # via @@ -640,10 +602,9 @@ pooch==1.8.2 portalocker==2.10.1 # via sacrebleu pqdm==0.2.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in prometheus-client==0.22.0 # via - # -c requirements/common.txt # opentelemetry-exporter-prometheus # ray propcache==0.2.0 @@ -654,7 +615,6 @@ proto-plus==1.26.1 # via google-api-core protobuf==6.33.6 # via - # -c requirements/common.txt # google-api-core # googleapis-common-protos # grpcio-reflection @@ -689,27 +649,25 @@ pycryptodomex==3.22.0 # via blobfile pydantic==2.12.0 # via - # -c requirements/common.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # albumentations # datamodel-code-generator - # fastapi - # gpt-oss # mistral-common # mteb - # openai-harmony # pydantic-extra-types # ray pydantic-core==2.41.1 # via pydantic -pydantic-extra-types==2.10.5 +pydantic-extra-types[pycountry]==2.10.5 # via mistral-common pygments==2.18.0 # via # pytest # rich -pyjwt==2.11.0 - # via msal +pyjwt[crypto]==2.11.0 + # via + # msal + # pyjwt pyparsing==3.2.0 # via matplotlib pyrate-limiter==4.4.0 @@ -720,7 +678,7 @@ pytablewriter==1.2.0 # via lm-eval pytest==9.1.0 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # buildkite-test-collector # genai-perf # pytest-asyncio @@ -732,19 +690,19 @@ pytest==9.1.0 # pytest-timeout # schemathesis pytest-asyncio==1.4.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-cov==6.3.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-forked==1.6.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-mock==3.14.0 # via genai-perf pytest-rerunfailures==14.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-shard==0.1.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in pytest-timeout==2.3.1 - # via -r requirements/test/cuda.in + # via -r test/cuda.in python-dateutil==2.9.0.post0 # via # botocore @@ -779,8 +737,8 @@ pyyaml==6.0.2 # vocos rapidfuzz==3.12.1 # via jiwer -ray==2.48.0 - # via -r requirements/test/cuda.in +ray[cgraph,default]==2.48.0 + # via -r test/cuda.in redis==5.2.0 # via tensorizer referencing==0.35.1 @@ -796,15 +754,12 @@ regex==2026.2.28 # transformers requests==2.32.3 # via - # -c requirements/common.txt # azure-core # buildkite-test-collector # datasets - # docker # evaluate # google-api-core # google-cloud-storage - # gpt-oss # lm-eval # mistral-common # msal @@ -832,8 +787,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 test/cuda.in runai-model-streamer-azure==0.15.7 # via runai-model-streamer runai-model-streamer-gcs==0.15.7 @@ -846,7 +801,6 @@ sacrebleu==2.4.3 # via lm-eval safetensors==0.7.0 # via - # -c requirements/common.txt # accelerate # open-clip-torch # peft @@ -854,7 +808,7 @@ safetensors==0.7.0 # timm # transformers schemathesis==4.21.6 - # via -r requirements/test/cuda.in + # via -r test/cuda.in scikit-image==0.25.2 # via albumentations scikit-learn==1.5.2 @@ -877,23 +831,17 @@ scipy==1.13.1 # statsmodels # vocos segmentation-models-pytorch==0.5.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in sentence-transformers==5.2.0 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # mteb -setuptools==77.0.3 - # via - # -c requirements/common.txt - # pytablewriter - # torch shellingham==1.5.4 # via # perceptron # typer six==1.16.0 # via - # -c requirements/common.txt # junit-xml # opencensus # python-dateutil @@ -908,7 +856,7 @@ sortedcontainers==2.4.0 # via hypothesis soundfile==0.12.1 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # genai-perf # librosa # mistral-common @@ -923,16 +871,11 @@ sqlalchemy==2.0.41 sqlitedict==2.1.0 # via lm-eval starlette==1.3.1 - # via - # -c requirements/common.txt - # fastapi - # starlette-testclient + # via starlette-testclient starlette-testclient==0.4.1 # via schemathesis statsmodels==0.14.4 # via genai-perf -structlog==25.4.0 - # via gpt-oss sympy==1.13.3 # via # einx @@ -942,43 +885,36 @@ tabledata==1.3.3 tabulate==0.9.0 # via sacrebleu tblib==3.1.0 - # via -r requirements/test/cuda.in + # via -r test/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 test/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 - # gpt-oss # lm-eval # mistral-common timm==1.0.17 # via - # -r requirements/test/cuda.in + # -r test/cuda.in # open-clip-torch # segmentation-models-pytorch tokenizers==0.22.2 # via - # -c requirements/common.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # transformers -torch==2.11.0+cu130 +torch==2.11.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # accelerate # bitsandbytes # encodec @@ -994,16 +930,14 @@ torch==2.11.0+cu130 # torchvision # vector-quantize-pytorch # vocos -torchaudio==2.11.0+cu130 +torchaudio==2.11.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # encodec # vocos -torchvision==0.26.0+cu130 +torchvision==0.26.0 # via - # -c requirements/cuda.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # open-clip-torch # segmentation-models-pytorch # timm @@ -1024,19 +958,18 @@ tqdm==4.67.3 # transformers transformers==5.5.3 # via - # -c requirements/common.txt - # -r requirements/test/cuda.in + # -r test/cuda.in # genai-perf # peft # sentence-transformers # transformers-stream-generator transformers-stream-generator==0.0.5 - # via -r requirements/test/cuda.in + # via -r test/cuda.in triton==3.6.0 # via torch tritonclient==2.64.0 - # via -r requirements/test/cuda.in -typepy==1.3.2 + # via -r test/cuda.in +typepy[datetime]==1.3.2 # via # dataproperty # pytablewriter @@ -1049,15 +982,12 @@ typer==0.15.2 # transformers typing-extensions==4.15.0 # via - # -c requirements/common.txt # aiosignal # albumentations # alembic # azure-core # azure-identity # azure-storage-blob - # chz - # fastapi # grpcio # huggingface-hub # librosa @@ -1080,27 +1010,22 @@ typing-extensions==4.15.0 # typer # typing-inspection typing-inspection==0.4.2 - # via - # fastapi - # pydantic + # via pydantic tzdata==2024.2 # via pandas urllib3==2.2.3 # via # blobfile # botocore - # docker # requests # responses # tritonclient -uvicorn==0.35.0 - # via gpt-oss vector-quantize-pytorch==1.21.2 - # via -r requirements/test/cuda.in + # via -r test/cuda.in virtualenv==20.31.2 # via ray vocos==0.1.0 - # via -r requirements/test/cuda.in + # via -r test/cuda.in wcwidth==0.2.13 # via ftfy werkzeug==3.1.3 diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index 726aad9a672b..795f1575816a 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 test/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 test/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 test/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 test/rocm.in bitsandbytes==0.49.2 - # via -r requirements/test/rocm.in + # via -r test/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 test/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 test/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.3.1 # via # black @@ -128,7 +121,7 @@ click==8.3.1 # uvicorn cloudpickle==3.1.2 # via - # -r requirements/test/../common.txt + # -r common.txt # tilelang colorama==0.4.6 # via @@ -139,12 +132,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 @@ -155,28 +146,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 +datamodel-code-generator==0.64.0 + # via -r test/rocm.in dataproperty==1.1.0 # via # pytablewriter # tabledata datasets==3.6.0 # via - # -r requirements/test/rocm.in + # -r test/rocm.in # evaluate # lm-eval # mteb decorator==5.2.1 # via librosa decord==0.6.0 - # via -r requirements/test/rocm.in + # via -r test/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 @@ -185,9 +184,7 @@ dill==0.3.8 # lm-eval # multiprocess diskcache==5.6.3 - # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # via -r common.txt distlib==0.4.0 # via virtualenv distro==1.9.0 @@ -196,16 +193,14 @@ 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 requirements/test/rocm.in + # -r common.txt + # -r test/rocm.in # encodec # vector-quantize-pytorch # vocos @@ -221,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 @@ -236,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 test/rocm.in filelock==3.25.2 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r common.txt # blobfile # datasets # huggingface-hub @@ -258,7 +248,7 @@ frozenlist==1.8.0 # via # aiohttp # aiosignal -fsspec==2025.3.0 +fsspec[http]==2025.3.0 # via # datasets # evaluate @@ -268,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 test/rocm.in genson==1.3.0 # via datamodel-code-generator google-api-core==2.30.0 @@ -297,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 test/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 test/rocm.in h11==0.16.0 # via # httpcore @@ -328,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 test/rocm.in # anthropic # fastapi # fastapi-cloud-cli @@ -382,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 test/rocm.in imageio==2.37.3 # via scikit-image importlib-metadata==8.7.1 @@ -394,7 +377,7 @@ inflect==7.5.0 iniconfig==2.3.0 # via pytest instanttensor==0.1.6 - # via -r requirements/test/rocm.in + # via -r test/rocm.in interegular==0.3.3 # via lm-format-enforcer isodate==0.7.2 @@ -413,7 +396,7 @@ jiter==0.14.0 # anthropic # openai jiwer==4.0.0 - # via -r requirements/test/rocm.in + # via -r test/rocm.in jmespath==1.1.0 # via # boto3 @@ -426,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 @@ -439,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 test/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 @@ -455,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 test/rocm.in +llguidance==1.7.5 ; platform_machine == "x86_64" or platform_machine == "arm64" or platform_machine == "aarch64" or platform_machine == "ppc64le" + # 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 test/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 @@ -477,7 +453,6 @@ loguru==0.7.3 lxml==6.0.2 # via # blobfile - # gpt-oss # sacrebleu mako==1.3.10 # via alembic @@ -489,27 +464,25 @@ markupsafe==3.0.3 # mako # werkzeug matplotlib==3.10.8 - # via -r requirements/test/rocm.in + # via -r test/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 test/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 @@ -527,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 test/rocm.in multidict==6.7.1 # via # aiohttp # yarl multiprocess==0.70.16 # via - # -r requirements/test/rocm.in + # -r test/rocm.in # datasets # evaluate mypy-extensions==1.1.0 @@ -548,20 +521,19 @@ 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 test/rocm.in numba==0.65.0 # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in + # -r test/rocm.in # librosa numpy==2.2.6 # via - # -r requirements/test/../common.txt - # -r requirements/test/rocm.in + # -r common.txt + # -r test/rocm.in # accelerate # albumentations # bitsandbytes @@ -611,43 +583,75 @@ 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 test/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 requirements/test/rocm.in + # -r common.txt + # -r test/rocm.in # albumentations # mistral-common openpyxl==3.1.5 - # via -r requirements/test/rocm.in + # via -r test/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 @@ -666,8 +670,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 @@ -678,9 +681,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 @@ -688,12 +689,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 @@ -724,7 +722,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 @@ -732,14 +730,14 @@ pathvalidate==3.3.1 patsy==1.0.2 # via statsmodels peft==0.18.1 - # via -r requirements/test/rocm.in + # via -r test/rocm.in perceptron==0.1.4 - # via -r requirements/test/rocm.in + # via -r test/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 @@ -757,7 +755,7 @@ platformdirs==4.3.6 # virtualenv plotly==6.6.0 # via - # -r requirements/test/rocm.in + # -r test/rocm.in # genai-perf pluggy==1.6.0 # via @@ -772,18 +770,15 @@ pooch==1.8.2 portalocker==3.2.0 # via sacrebleu pqdm==0.2.0 - # via -r requirements/test/rocm.in + # via -r test/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 @@ -792,8 +787,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 @@ -803,7 +797,7 @@ protobuf==6.33.6 # tensorizer psutil==7.2.2 # via - # -r requirements/test/../common.txt + # -r common.txt # accelerate # peft # tensorizer @@ -811,7 +805,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 @@ -823,25 +817,23 @@ 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 requirements/test/rocm.in + # -r common.txt + # -r test/rocm.in # albumentations # anthropic # compressed-tensors # datamodel-code-generator # fastapi # fastapi-cloud-cli - # gpt-oss # lm-format-enforcer # mcp # mistral-common @@ -855,7 +847,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 @@ -867,7 +859,7 @@ pygments==2.19.2 # via # pytest # rich -pyjwt==2.12.1 +pyjwt[crypto]==2.12.1 # via # mcp # msal @@ -881,7 +873,7 @@ pytablewriter==1.2.1 # via lm-eval pytest==9.1.0 # via - # -r requirements/test/rocm.in + # -r test/rocm.in # buildkite-test-collector # genai-perf # pytest-asyncio @@ -893,19 +885,19 @@ pytest==9.1.0 # pytest-timeout # schemathesis pytest-asyncio==1.4.0 - # via -r requirements/test/rocm.in + # via -r test/rocm.in pytest-cov==6.3.0 - # via -r requirements/test/rocm.in + # via -r test/rocm.in pytest-forked==1.6.0 - # via -r requirements/test/rocm.in + # via -r test/rocm.in pytest-mock==3.15.1 # via genai-perf pytest-rerunfailures==14.0 - # via -r requirements/test/rocm.in + # via -r test/rocm.in pytest-shard==0.1.2 - # via -r requirements/test/rocm.in + # via -r test/rocm.in pytest-timeout==2.3.1 - # via -r requirements/test/rocm.in + # via -r test/rocm.in python-dateutil==2.9.0.post0 # via # botocore @@ -919,7 +911,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 @@ -936,7 +928,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 @@ -954,15 +946,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 test/rocm.in # jiwer -ray==2.54.0 - # via -r requirements/test/rocm.in +ray[cgraph,default]==2.54.0 + # via -r test/rocm.in redis==7.3.0 # via tensorizer referencing==0.37.0 @@ -971,7 +961,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 @@ -979,16 +969,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 @@ -1022,10 +1009,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 test/rocm.in runai-model-streamer-azure==0.15.7 # via runai-model-streamer runai-model-streamer-gcs==0.15.7 @@ -1038,8 +1023,7 @@ sacrebleu==2.6.0 # via lm-eval safetensors==0.7.0 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt + # -r common.txt # accelerate # open-clip-torch # peft @@ -1047,7 +1031,7 @@ safetensors==0.7.0 # timm # transformers schemathesis==4.21.6 - # via -r requirements/test/rocm.in + # via -r test/rocm.in scikit-image==0.26.0 # via albumentations scikit-learn==1.8.0 @@ -1071,25 +1055,17 @@ scipy==1.17.1 # statsmodels # vocos segmentation-models-pytorch==0.5.0 - # via -r requirements/test/rocm.in + # via -r test/rocm.in sentence-transformers==5.3.0 # via - # -r requirements/test/rocm.in + # -r test/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 @@ -1098,8 +1074,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 @@ -1115,7 +1089,7 @@ sortedcontainers==2.4.0 # via hypothesis soundfile==0.13.1 # via - # -r requirements/test/rocm.in + # -r test/rocm.in # genai-perf # librosa # mistral-common @@ -1133,8 +1107,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 @@ -1145,8 +1118,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 @@ -1158,52 +1129,71 @@ tabledata==1.3.4 tabulate==0.10.0 # via sacrebleu tblib==3.1.0 - # via -r requirements/test/rocm.in + # via -r test/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 test/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 test/rocm.in timm==1.0.17 # via - # -c requirements/rocm.txt - # -r requirements/test/rocm.in + # -r test/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 test/rocm.in # transformers +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 @@ -1221,9 +1211,8 @@ tqdm==4.67.3 # transformers transformers==5.5.3 # via - # -c requirements/common.txt - # -r requirements/test/../common.txt - # -r requirements/test/rocm.in + # -r common.txt + # -r test/rocm.in # compressed-tensors # genai-perf # peft @@ -1231,12 +1220,16 @@ transformers==5.5.3 # transformers-stream-generator # xgrammar transformers-stream-generator==0.0.5 - # via -r requirements/test/rocm.in + # via -r test/rocm.in +triton==3.7.1 + # via + # torch + # xgrammar tritonclient==2.66.0 - # via -r requirements/test/rocm.in + # via -r test/rocm.in typeguard==4.5.1 # via inflect -typepy==1.3.4 +typepy[datetime]==1.3.4 # via # dataproperty # pytablewriter @@ -1251,8 +1244,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 @@ -1262,7 +1254,7 @@ typing-extensions==4.15.0 # azure-core # azure-identity # azure-storage-blob - # chz + # blake3 # fastapi # grpcio # huggingface-hub @@ -1303,29 +1295,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 test/rocm.in virtualenv==21.2.0 # via ray vocos==0.1.0 - # via -r requirements/test/rocm.in + # via -r test/rocm.in watchfiles==1.1.1 # via - # -r requirements/test/../common.txt + # -r common.txt # uvicorn wcwidth==0.6.0 # via ftfy @@ -1337,10 +1327,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 @@ -1351,28 +1339,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