diff --git a/requirements/tpu.txt b/requirements/tpu.txt
index b63993ba1ee4..16c0ad3ecb13 100644
--- a/requirements/tpu.txt
+++ b/requirements/tpu.txt
@@ -9,6 +9,7 @@ wheel
 jinja2>=3.1.6
 ray[default]
 ray[data]
+setuptools==78.1.0
 
 # Install torch_xla
 --pre