Skip to content

Add conversational serving - #165

Merged
jonatanklosko merged 1 commit into
mainfrom
jk-conversational
Feb 24, 2023
Merged

Add conversational serving#165
jonatanklosko merged 1 commit into
mainfrom
jk-conversational

Conversation

@jonatanklosko

@jonatanklosko jonatanklosko commented Feb 24, 2023

Copy link
Copy Markdown
Member

image

@seanmor5

Copy link
Copy Markdown
Contributor

I'm not sure, but I think I'm gonna go to sleep

I felt that

Comment thread lib/bumblebee/text.ex
Each call to serving returns the conversation history, which can be
fed into the next run to maintain the context.

Note that either `:max_new_tokens` or `:max_length` must be specified.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

I think max_length is missing from the options below?

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

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

It's intentional, we do say "Also accepts all the other options of Bumblebee.Text.Generation.build_generate/3" and both :max_new_tokens and :max_length are there. I think :max_new_tokens is common enough that it makes sense to list it here, but it should be preferred over :max_length, and so I didn't include both.

@jonatanklosko
jonatanklosko merged commit 8b6a8bf into main Feb 24, 2023
@jonatanklosko
jonatanklosko deleted the jk-conversational branch February 24, 2023 20:24
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.

3 participants