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

Track the full probability chain along each beam instead of only the final string and accumulated probability mass #1

Open
the-crypt-keeper opened this issue May 27, 2024 · 1 comment
Labels
enhancement New feature or request

Comments

@the-crypt-keeper
Copy link
Owner

This is the biggest corner that I cut in this implementation and its blocking some nice stuff:

  • intelligent pruning of suggestion beams when there are too many. auto-raising the Multiplier and use the new thresholds to cut off beams until we have the number we want.
  • currently we do a post-processing greedy string prefix finding step prior to visualization to recover the forked streams, this would no longer be necessary if we could access the underlying streams directly. this would enable use of colors to indicate high/low probability tokens in the visualization, but at the expense of bleeding through a lot more tokenization details then the current visualization does so maybe its wise to keep both methods around.
@the-crypt-keeper the-crypt-keeper added the enhancement New feature or request label May 27, 2024
@the-crypt-keeper the-crypt-keeper changed the title Track the full probability chain along each beam instead of only the final string and accumulated proability mass Track the full probability chain along each beam instead of only the final string and accumulated probability mass May 27, 2024
@the-crypt-keeper
Copy link
Owner Author

This is now much easier with the new iterative BFS search, but still not done 😄

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

No branches or pull requests

1 participant