Add a comment

Signed-off-by: Huy Do <huydhn@gmail.com>

Author: huydhn

requirements/test.in

@@ -28,7 +28,7 @@ timm >=1.0.17 # required for internvl and gemma3n-mm test
 torch==2.9.0
 torchaudio==2.9.0
 torchvision==0.24.0
-torchao==0.13.0
+torchao==0.13.0 # for quantization tests
 transformers_stream_generator # required for qwen-vl test
 matplotlib # required for qwen-vl test
 mistral_common[image,audio] >= 1.8.2 # required for voxtral test

requirements/test.txt

@@ -1,5 +1,5 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match --torch-backend cu128
+#    uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match
 absl-py==2.1.0
     # via rouge-score
 accelerate==1.0.1
@@ -570,14 +570,16 @@ nvidia-cusparse-cu12==12.5.8.93
     #   torch
 nvidia-cusparselt-cu12==0.7.1
     # via torch
-nvidia-nccl-cu12==2.27.3
+nvidia-nccl-cu12==2.27.5
     # via torch
 nvidia-nvjitlink-cu12==12.8.93
     # via
     #   nvidia-cufft-cu12
     #   nvidia-cusolver-cu12
     #   nvidia-cusparse-cu12
     #   torch
+nvidia-nvshmem-cu12==3.3.20
+    # via torch
 nvidia-nvtx-cu12==12.8.90
     # via torch
 omegaconf==2.3.0
@@ -979,7 +981,6 @@ setuptools==77.0.3
     #   lightning-utilities
     #   pytablewriter
     #   torch
-    #   triton
 shapely==2.1.1
     # via
     #   geopandas
@@ -1078,7 +1079,7 @@ tomli==2.2.1
     # via schemathesis
 tomli-w==1.2.0
     # via schemathesis
-torch==2.8.0+cu128
+torch==2.9.0+cu128
     # via
     #   -r requirements/test.in
     #   accelerate
@@ -1107,7 +1108,9 @@ torch==2.8.0+cu128
     #   torchvision
     #   vector-quantize-pytorch
     #   vocos
-torchaudio==2.8.0+cu128
+torchao==0.13.0+cu128
+    # via -r requirements/test.in
+torchaudio==2.9.0+cu128
     # via
     #   -r requirements/test.in
     #   encodec
@@ -1120,7 +1123,7 @@ torchmetrics==1.7.4
     #   pytorch-lightning
     #   terratorch
     #   torchgeo
-torchvision==0.23.0+cu128
+torchvision==0.24.0+cu128
     # via
     #   -r requirements/test.in
     #   lightly
@@ -1161,7 +1164,7 @@ transformers==4.55.2
     #   transformers-stream-generator
 transformers-stream-generator==0.0.5
     # via -r requirements/test.in
-triton==3.4.0
+triton==3.5.0
     # via torch
 tritonclient==2.51.0
     # via