From 1eaf64becf4c12088a5e95851e3f2623e02b1e62 Mon Sep 17 00:00:00 2001 From: Haoliang Zhang Date: Thu, 6 Jun 2024 12:59:35 -0700 Subject: [PATCH] update tf-nightly version so that models w/ external cache can be converted. (#37) --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index b925ce67..8aa36750 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ numpy tabulate safetensors --pre -tf-nightly-cpu>=2.17.0.dev20240604 +tf-nightly-cpu>=2.17.0.dev20240606 -f https://download.pytorch.org/whl/nightly/torch_nightly.html torch==2.4.0.dev20240429+cpu -f https://download.pytorch.org/whl/nightly/torch_nightly.html @@ -12,4 +12,4 @@ torchvision==0.19.0.dev20240429+cpu torchaudio==2.2.0.dev20240429+cpu torch_xla@https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly+20240429-cp311-cp311-linux_x86_64.whl; python_version=="3.11" torch_xla@https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly+20240429-cp310-cp310-linux_x86_64.whl; python_version=="3.10" -torch_xla@https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly+20240429-cp39-cp39-linux_x86_64.whl; python_version=="3.9" \ No newline at end of file +torch_xla@https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly+20240429-cp39-cp39-linux_x86_64.whl; python_version=="3.9"