Import error

[nothingness6]

Hi, I just ran into the import error below:

`SystemError Traceback (most recent call last)

in
8 from PIL import Image
9
---> 10 from stable_diffusion_videos import FlaxStableDiffusionWalkPipeline, Interface
11
12 pipeline, params = FlaxStableDiffusionWalkPipeline.from_pretrained(

12 frames

/usr/local/lib/python3.8/dist-packages/numba/np/ufunc/decorators.py in
1 import inspect
2
----> 3 from numba.np.ufunc import _internal
4 from numba.np.ufunc.parallel import ParallelUFuncBuilder, ParallelGUFuncBuilder
5

SystemError: initialization of _internal failed without raising an exception`

I have no idea what to do. Please give me a solution.
Thank you.

[nateraw]

Can you please refer to solution mentioned in #168 and see if it fixes your problem? comment here. Just don't reinstall jax, then remove the param passed to setup (so should just be jax.tools.colab_tpu.setup_tpu() instead of one with the build string passed).

[nateraw]

Please let me know if it does - I'll hotfix asap

[nothingness6]

Hi, nateraw.
I appreciate it. Unfortunately, I just faced to another error from the start (Set up JAX). So, I couldn't go further. Here is the error message:

`KeyError Traceback (most recent call last)

in
4
5 import jax.tools.colab_tpu
----> 6 jax.tools.colab_tpu.setup_tpu('tpu_driver_20221011')
7
8 get_ipython().system('pip install flax diffusers transformers ftfy')

1 frames

/usr/lib/python3.8/os.py in getitem(self, key)
673 except KeyError:
674 # raise KeyError with the original key value
--> 675 raise KeyError(key) from None
676 return self.decodevalue(value)
677

KeyError: 'COLAB_TPU_ADDR'`