change base image

Author: lanluo-nvidia

.github/scripts/filter-matrix.py

@@ -28,11 +28,15 @@ def main(args: List[str]) -> None:
         if all(
             [
                 item["python_version"] not in disabled_python_versions,
-                # only test cu128 for aarch64 for now for quick CI
-                item["desired_cuda"] == "cu128",
             ]
         ):
-            filtered_includes.append(item)
+            if item["gpu_arch_type"] == "cuda-aarch64":
+                # for aarch64 only test cu128 for now for quick CI
+                if item["desired_cuda"] == "cu128":
+                    item["container_image"] = "sameli/manylinux_2_34_x86_64_cuda_12.8"
+                    filtered_includes.append(item)
+            else:
+                filtered_includes.append(item)
     filtered_matrix_dict = {}
     filtered_matrix_dict["include"] = filtered_includes
     print(json.dumps(filtered_matrix_dict))

tests/py/requirements.txt

@@ -8,6 +8,6 @@ pytest>=8.2.1
 pytest-xdist>=3.6.1
 pyyaml
 timm>=1.0.3
-transformers==4.40.2
-nvidia-modelopt[deploy,hf,torch]~=0.17.0
+transformers==4.49.0
+nvidia-modelopt[all]~=0.27.0; python_version >'3.9' and python_version <'3.13'
 --extra-index-url https://pypi.nvidia.com