vicgalle/gpt-j-api

AssertionError

Opened this issue · 0 comments

For installation cat requirements.txt | xargs -n 1 pip3 install

I don't run intro previous errors with latest
Name: jax Version: 0.3.11
Name: jaxlib Version: 0.3.11
suggestion to update requirements.txt

jax and jaxlib Building from source

was run into this error before getting experience using streamlit run but this is not python3 serve.py and now I get the same error

WARNING:absl:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
Traceback (most recent call last):
  File "/Users/macos/Desktop/AI/serve.py", line 43, in <module>
    devices = np.array(jax.devices()).reshape(mesh_shape)
ValueError: cannot reshape array of size 1 into shape (0,8)