Inference Notebook Does Not Work

[sajeedmehrab]

[Boris Dayma]

I think it should now work. Can you confirm?

[colabtracemedia]

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 jnp

RuntimeError: jaxlib is version 0.3.25, but this version of jax requires version >= 0.4.2.

[Boris Dayma]

Ouch, this issue only appeared yesterday 😅
New attempt, I pushed a new fix (now pinning jax & flax as well)!

[colabtracemedia]

Great, thank you, Inference Notebook now fixed!

[colabtracemedia]

Hi, it looks like there's now an error with the version of jaxlib when running the Inference Notebook:

[robitec97]

Notebook is not working

[colabtracemedia]

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

[marburps]

Loading VQGAN takes forever (notebook just keeps loading endlessly)

[Boris Dayma]

I think the newest release should now directly work