update modules

Author: jiazhihao

deps/tokenizers-cpp

@@ -18,6 +18,11 @@ get_cuda_docker_image() {
       ubuntu_version="22.04"
   fi
 
+  # Downgrade to 22.04 for now
+  if [[ "$ubuntu_version" == "24.04" ]]; then
+    ubuntu_version="22.04"
+  fi
+
   # Determine CUDA version.
   # If the environment variable 'cuda_version' is set (in "<major>.<minor>" format), use that.
   # Otherwise, use nvidia-smi to extract the CUDA version.

docker/build.sh

@@ -19,9 +19,9 @@ install_pytorch() {
   echo "Attempting to install PyTorch with CUDA ${major}.${minor} support..."
 
   # Run dry-run first
-  if pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu"${major}${minor}" --dry-run; then
+  if pip3 install "torch<=2.7.1" torchvision torchaudio --index-url https://download.pytorch.org/whl/cu"${major}${minor}" --dry-run; then
     echo "Dry-run succeeded, proceeding with actual installation..."
-    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu"${major}${minor}"
+    pip3 install "torch<=2.7.1" torchvision torchaudio --index-url https://download.pytorch.org/whl/cu"${major}${minor}"
     return 0
   else
     echo "Dry-run failed for CUDA ${major}.${minor}."