diff --git a/infra/tpu-pytorch-releases/artifacts.auto.tfvars b/infra/tpu-pytorch-releases/artifacts.auto.tfvars index cbec743e3806..39a1451968b3 100644 --- a/infra/tpu-pytorch-releases/artifacts.auto.tfvars +++ b/infra/tpu-pytorch-releases/artifacts.auto.tfvars @@ -49,6 +49,7 @@ xrt_versioned_builds = [ # Built on push to specific tag. versioned_builds = [ + # Remove libtpu from PyPI builds { git_tag = "v2.1.0" pytorch_git_rev = "v2.1.0" @@ -56,6 +57,14 @@ versioned_builds = [ accelerator = "tpu" bundle_libtpu = "0" }, + { + git_tag = "v2.1.0" + pytorch_git_rev = "v2.1.0" + package_version = "2.1.0" + accelerator = "tpu" + python_version = "3.9" + bundle_libtpu = "0" + }, { git_tag = "v2.1.0" pytorch_git_rev = "v2.1.0" @@ -64,6 +73,7 @@ versioned_builds = [ python_version = "3.10" bundle_libtpu = "0" }, + # Bundle libtpu for Kaggle { git_tag = "v2.1.0" pytorch_git_rev = "v2.1.0"