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 question answering serving #157

Merged
merged 11 commits into from
Feb 6, 2023
Merged

Conversation

blackeuler
Copy link
Contributor

@blackeuler blackeuler commented Feb 2, 2023

I did some poking around, and I think this is how question answering would be added. I know I need to clean up the docs and some other things like testing but feedback would be appreciated on the overall approach. Thanks so much for the excellent library.

Copy link
Member

@jonatanklosko jonatanklosko left a comment

Choose a reason for hiding this comment

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

Thanks for working on this! I dropped some high-level feedback :)

lib/bumblebee/text.ex Outdated Show resolved Hide resolved
lib/bumblebee/text.ex Outdated Show resolved Hide resolved
lib/bumblebee/text/question_answering.ex Outdated Show resolved Hide resolved
lib/bumblebee/text/question_answering.ex Outdated Show resolved Hide resolved
lib/bumblebee/text/question_answering.ex Outdated Show resolved Hide resolved
lib/bumblebee/text.ex Show resolved Hide resolved
@jonatanklosko
Copy link
Member

@blackeuler thanks :) I pushed some adjustments and docs, I think we are all good now.

There are some options we may want to support going forward, in particular top_k and document_stride (which will likely require changes to the tokenizers bindings). But I think we can add those as needed.

@jonatanklosko jonatanklosko changed the title Add Question Answering Task Add question answering serving Feb 6, 2023
@jonatanklosko jonatanklosko merged commit f57f2d2 into elixir-nx:main Feb 6, 2023
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