Bump Ray version to 2.54.1 (#5864)

Author: jinyan-li1

.github/config/ray-ec2-cpu.yml

@@ -9,7 +9,7 @@ image:
 # Build configuration
 common:
   framework: "ray"
-  framework_version: "2.54.0"
+  framework_version: "2.54.1"
   job_type: "inference"
   python_version: "py313"
   os_version: "amzn2023"

.github/config/ray-ec2-gpu.yml

@@ -9,7 +9,7 @@ image:
 # Build configuration
 common:
   framework: "ray"
-  framework_version: "2.54.0"
+  framework_version: "2.54.1"
   job_type: "inference"
   python_version: "py313"
   cuda_version: "cu129"

.github/config/ray-sagemaker-cpu.yml

@@ -9,7 +9,7 @@ image:
 # Build configuration
 common:
   framework: "ray"
-  framework_version: "2.54.0"
+  framework_version: "2.54.1"
   job_type: "inference"
   python_version: "py313"
   os_version: "amzn2023"

.github/config/ray-sagemaker-gpu.yml

@@ -9,7 +9,7 @@ image:
 # Build configuration
 common:
   framework: "ray"
-  framework_version: "2.54.0"
+  framework_version: "2.54.1"
   job_type: "inference"
   python_version: "py313"
   cuda_version: "cu129"

docker/ray/Dockerfile.cpu

@@ -56,7 +56,7 @@ LABEL dlc_minor_version="0"
 
 ARG PYTHON="python"
 ARG FRAMEWORK="ray"
-ARG FRAMEWORK_VERSION="2.54.0"
+ARG FRAMEWORK_VERSION="2.54.1"
 ARG CONTAINER_TYPE="inference"
 
 # Copy Python installation and venv from builder

docker/ray/Dockerfile.gpu

@@ -79,7 +79,7 @@ LABEL dlc_minor_version="0"
 
 ARG PYTHON="python"
 ARG FRAMEWORK="ray"
-ARG FRAMEWORK_VERSION="2.54.0"
+ARG FRAMEWORK_VERSION="2.54.1"
 ARG CONTAINER_TYPE="inference"
 
 # Enable video capability to mount NVENC/NVDEC driver libraries

scripts/ray/pyproject.toml

@@ -13,7 +13,7 @@ dependencies = [
     "pandas==3.0.1",
     "pillow==12.1.1",
     "pyyaml==6.0.3",
-    "ray[serve]==2.54.0",
+    "ray[serve]==2.54.1",
     "scikit-learn==1.8.0",
     "soundfile==0.13.1",
     "torch==2.10.0",