Skip to content

Commit

Permalink
update tf-nightly version so that models w/ external cache can be con…
Browse files Browse the repository at this point in the history
…verted. (#37)
  • Loading branch information
haozha111 authored Jun 6, 2024
1 parent 382a02c commit 1eaf64b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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"
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"

0 comments on commit 1eaf64b

Please sign in to comment.