New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fix readme jax version installation, reset args bug #64

Merged

youliangtan merged 1 commit into main from fix-installation-bugs

Jun 22, 2024

Member

youliangtan commented Jun 22, 2024

fix jax installation instruction:
- If install jax[cuda12] > v0.4.29 will get sharding error due to api changes Update README.md #61, yet =0.4.25 still giving weird seg fault. Thus revert to 0.4.28
fix chunking reset missing args parsing


          fix readme jax version installation, reset args bug

2fdb2f6

Signed-off-by: youliangtan <[email protected]>

youliangtan marked this pull request as ready for review

June 22, 2024 04:22

youliangtan merged commit 0187476 into main

1 check passed

youliangtan deleted the fix-installation-bugs branch

June 22, 2024 04:22

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet