-
Notifications
You must be signed in to change notification settings - Fork 412
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Set up JAX ERROR #199
Comments
This order worked for me in a colab. You basically have to turn time back to November, 2022, before the jax0.4 versions. I did not even have to restart like it asks.
|
Note that the jax/tpu version can not deal with negative prompts. I ended up going back to the regular gpu version, which also seemed faster. |
Thank you @aduchon for the snippet! I'll look into this and update the colab accordingly. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
The TPU Version wont start.
I can set up JAX with "jax==0.3.25 jaxlib==0.3.25", but then interface and Walk! wont work. Could you have a look at your TPU version with you basic settings?
The text was updated successfully, but these errors were encountered: