Docker AttributeError: module 'jaxlib.pocketfft' has no attribute 'pocketfft

[cdmichaelb]

`Traceback (most recent call last):

File "app.py", line 18, in

from dalle_model import DalleModel

File "/app/dalle_model.py", line 5, in

import jax

File "/usr/local/lib/python3.8/dist-packages/jax/init.py", line 116, in

from .experimental.maps import soft_pmap as soft_pmap

File "/usr/local/lib/python3.8/dist-packages/jax/experimental/maps.py", line 26, in

from .. import numpy as jnp

File "/usr/local/lib/python3.8/dist-packages/jax/numpy/init.py", line 19, in

from . import fft as fft

File "/usr/local/lib/python3.8/dist-packages/jax/numpy/fft.py", line 17, in

from jax._src.numpy.fft import (

File "/usr/local/lib/python3.8/dist-packages/jax/_src/numpy/fft.py", line 19, in

from jax import lax

File "/usr/local/lib/python3.8/dist-packages/jax/lax/init.py", line 332, in

from jax._src.lax.fft import (

File "/usr/local/lib/python3.8/dist-packages/jax/_src/lax/fft.py", line 145, in

xla.backend_specific_translations['cpu'][fft_p] = pocketfft.pocketfft

AttributeError: module 'jaxlib.pocketfft' has no attribute 'pocketfft'`

[hibby50]

Same issue here

[cdmichaelb]

@hibby50

Fixed with #79

[JoshWarn]

Still seem to be having this issue while working with MEGA.

[cdmichaelb]

Still seem to be having this issue while working with MEGA.

Check #79

[Larry-Schultz]

I manually changed my Dockerfile to replicate the changes in #79 and this issue still happens on startup.

EDIT: Looks like I forgot to do docker-compose build after changing the Dockerfile.

After I implemented the changes in PR #79, ran docker-compose build, then ran docker-compose up everything started working.