diff --git a/requirements/tpu.txt b/requirements/tpu.txt index 7bb77c4a99636..7ea239b48ea26 100644 --- a/requirements/tpu.txt +++ b/requirements/tpu.txt @@ -11,6 +11,7 @@ ray[default] ray[data] setuptools==78.1.0 nixl==0.3.0 +tpu_info==0.4.0 # Install torch_xla --pre