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

Expose n_vocab, n_embed, n_layer to the Python interface #118

Merged
merged 1 commit into from Jul 18, 2023

Conversation

mczk77
Copy link
Contributor

@mczk77 mczk77 commented Jul 17, 2023

This patch exposes the rwkv_get_n_vocab, rwkv_get_n_embed, rwkv_get_n_layer functions to the Python interface through the n_vocab, n_embed, n_layer properties of the RWKVSharedLibrary class.

@saharNooby
Copy link
Collaborator

CC @LoganDark

@LoganDark
Copy link
Contributor

Logan isn't here right now so can't actually offer his opinion, but it looks fine to me!

@saharNooby
Copy link
Collaborator

Just was tagging you since you are working on Python API rewrite (AFAIK) :)

@saharNooby saharNooby merged commit 25ee75e into RWKV:master Jul 18, 2023
12 checks passed
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.

None yet

3 participants