AssertionError
Opened this issue · 0 comments
Deleted user commented
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)