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

[BPE PR 2.4] BPE Class + Merge utilities #7774

Merged
merged 69 commits into from
Jun 27, 2023

Conversation

pforderique
Copy link
Contributor

This PR includes the base BPE class implementation including private merge functions that will be needed for the tokenize() function.

This PR only includes one test for the BPE setup configuration and no test suite for the merge functions since they are private.

pforderique and others added 30 commits June 14, 2023 21:46
Co-authored-by: Matthew Soulanille <matthew@soulanille.net>
Co-authored-by: Matthew Soulanille <matthew@soulanille.net>
tfjs-layers/src/layers/nlp/tokenizers.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers_utils.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers_utils.ts Outdated Show resolved Hide resolved
tfjs-layers/src/layers/nlp/tokenizers.ts Outdated Show resolved Hide resolved
@pforderique
Copy link
Contributor Author

@mattsoulanille comments have been resolved. Lmk if you have any more suggestions! thanks!

Copy link
Member

@mattsoulanille mattsoulanille left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@mattsoulanille
Copy link
Member

@Linchenn Please take a look. Thanks!

Copy link
Collaborator

@Linchenn Linchenn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Overall LGTM! Thanks!

tfjs-layers/src/layers/nlp/tokenizers.ts Show resolved Hide resolved
@pforderique pforderique merged commit e7082b4 into tensorflow:master Jun 27, 2023
2 checks passed
@pforderique pforderique deleted the bpe-merges branch June 27, 2023 23:23
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