Unable to initialize backend 'tpu': UNIMPLEMENTED: LoadPjrtPlugin is not implemented on windows yet. (set JAX_PLATFORMS='' to automatically choose an available backend)

openai / grok

MIT License

4.06k stars 506 forks source link

Open TeaMaster1337 opened 3 months ago

TeaMaster1337 commented 3 months ago

How to fix that?

ngovanuc commented 2 months ago

import os os.environ['JAX_PLATFORMS'] = ''

add the above code. If you're using Jupyter notebook try to run all cells in one time.