Skip to content

Commit

Permalink
fix jax cuda requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
@sampathweb
sampathweb committed Sep 27, 2024
1 parent 861c614 commit 2ed8c4c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion requirements-jax-cuda.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ torch>=2.1.0
torchvision>=0.16.0

# Jax with cuda support.
jax[cuda12]==0.4.29
jax[cuda12]==0.4.28
flax

-r requirements-common.txt

0 comments on commit 2ed8c4c

Please sign in to comment.