diff --git a/tools/ray_tpu/src/serve/requirements.txt b/tools/ray_tpu/src/serve/requirements.txt
index c62f0eb1..4058a089 100644
--- a/tools/ray_tpu/src/serve/requirements.txt
+++ b/tools/ray_tpu/src/serve/requirements.txt
@@ -1,3 +1,3 @@
 fastapi
-ray[serve]==2.5.1
+ray[serve]==2.54.0
 pillow
\ No newline at end of file