Skip to content

Commit

Permalink
find the working date
Browse files Browse the repository at this point in the history
  • Loading branch information
zpcore committed Jan 27, 2025
1 parent abe9bbb commit afa2310
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,10 +65,12 @@
base_dir = os.path.dirname(os.path.abspath(__file__))

USE_NIGHTLY = True # whether to use nightly or stable libtpu and jax
_date = '20250121'

_date = '20250105'
_libtpu_version = '0.0.8'
_jax_version = '0.5.1'
_jaxlib_version = '0.5.1'
_jax_version = '0.4.39'
_jaxlib_version = '0.4.39'

_libtpu_wheel_name = f'libtpu-{_libtpu_version}'
_libtpu_storage_directory = 'libtpu-lts-releases'

Expand Down

0 comments on commit afa2310

Please sign in to comment.