You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched ac
8000
counts on another tab or window. Reload to refresh your session.Dismiss alert
Adding jaxlib and cuda version to setup.cfg install_requires works for Google Colab Inference notebook: jaxlib@https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.3.25+cuda11.cudnn82-cp39-cp39-manylinux2014_x86_64.whl
Activity
borisdayma commentedon Feb 8, 2023
I think it should now work. Can you confirm?
colabtracemedia commentedon Feb 9, 2023
Hi, thanks for the update. I now get this error in the Colab Inference Notebook when jax is imported:
import jax import jax.numpy as jnpRuntimeError: jaxlib is version 0.3.25, but this version of jax requires version >= 0.4.2.
borisdayma commentedon Feb 10, 2023
Ouch, this issue only appeared yesterday 😅
New attempt, I pushed a new fix (now pinning jax & flax as well)!
colabtracemedia commentedon Feb 12, 2023
Great, thank you, Inference Notebook now fixed!
colabtracemedia commentedon Mar 4, 2023
Hi, it looks like there's now an error with the version of jaxlib when running the Inference Notebook:
robitec97 commentedon Mar 8, 2023
Notebook is not working
colabtracemedia commentedon Mar 14, 2023
Adding jaxlib and cuda version to setup.cfg install_requires works for Google Colab Inference notebook:
jaxlib@https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.3.25+cuda11.cudnn82-cp39-cp39-manylinux2014_x86_64.whlmarburps commentedon Apr 19, 2023
Loading VQGAN takes forever (notebook just keeps loading endlessly)
borisdayma commentedon Aug 22, 2023
I think the newest release should now directly work