Skip to content
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

Add Llama 3 tutorial (porting HF model to JAX) #133

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

windmaple
Copy link
Contributor

This is re-submission of PR #119.

This is re-submission of PR#119.
@windmaple
Copy link
Contributor Author

windmaple commented Dec 17, 2024

jax-ai-stack/docs/source/JAX_porting_HF_model.ipynb:100002: WARNING: skipping unknown output mime type: application/vnd.google.colaboratory.intrinsic+json [mystnb.unknown_mime_type] [mystnb.unknown_mime_type]

Looks like the build system does not like some codelab artifacts. @jakevdp any idea on how to make it happy?

@windmaple
Copy link
Contributor Author

Turns out this is generated by HF widget when downloading the weights. Can we suppress this?

Screenshot 2024-12-17 at 14 41 18

@jakevdp
Copy link
Collaborator

jakevdp commented Dec 18, 2024

You may be able to use the hide-output tag to avoid the problematic outputs

@jakevdp jakevdp self-assigned this Dec 18, 2024
@windmaple
Copy link
Contributor Author

windmaple commented Dec 19, 2024

Ended up removing that particular cell's output manually. Should work now.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants