sanchit-gandhi/whisper-jax

ERROR: Could not find a version that satisfies the requirement jaxlib==0.4.5 (from versions: 0.4.18, 0.4.19, 0.4.20, 0.4.21)

alcarazolabs opened this issue · 1 comments

Hi dear, I have python 3.9 in my device and according to the readme to run whisper-jax we need jax==0.4.5
however when I tried to install this two packages (jax and jaxlib) I'm getting this errores:

ERROR: Could not find a version that satisfies the requirement jaxlib==0.4.5 (from versions: 0.4.18, 0.4.19, 0.4.20, 0.4.21)
ERROR: No matching distribution found for jaxlib==0.4.5

I don't know why happens this.
https://pypi.org/project/jax/0.4.5/

Any idea to fix this problem, I will appreciate it. Thanks

Ok I just update jax and jaxlib and now worked:

$ pip install jax jaxlib
$ pip install --upgrade jax jaxlib flax