Not able to install install-heim-extras.sh for heim leaderboard

[snehith3195]

Hi All,

I am try to create replicate leaderboard to text to image models(HEIM) in colab but not able but getting the following error when trying to read the "install-heim-extras.sh" file:
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
chex 0.1.87 requires jax>=0.4.27, but you have jax 0.3.25 which is incompatible.
chex 0.1.87 requires jaxlib>=0.4.27, but you have jaxlib 0.3.25+cuda11.cudnn805 which is incompatible.
flax 0.6.11 requires jax>=0.4.2, but you have jax 0.3.25 which is incompatible.
optax 0.2.3 requires jax>=0.4.27, but you have jax 0.3.25 which is incompatible.
optax 0.2.3 requires jaxlib>=0.4.27, but you have jaxlib 0.3.25+cuda11.cudnn805 which is incompatible.
orbax-checkpoint 0.6.4 requires jax>=0.4.26, but you have jax 0.3.25 which is incompatible.

[yifanmai]

Could you try running the following and see if it fixes things for you?

pip install --upgrade jaxlib==0.4.30