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

[V1][Help Wanted] Porting missing sampling parameters to V1 #13058

Open
5 of 7 tasks
WoosukKwon opened this issue Feb 10, 2025 · 9 comments · Fixed by #13191
Open
5 of 7 tasks

[V1][Help Wanted] Porting missing sampling parameters to V1 #13058

WoosukKwon opened this issue Feb 10, 2025 · 9 comments · Fixed by #13191
Labels

Comments

@WoosukKwon
Copy link
Collaborator

WoosukKwon commented Feb 10, 2025

Anything you want to discuss about vllm.

To switch the engine from V0 to V1, we need to comprehensively support the sampling parameters in https://github.com/vllm-project/vllm/blob/main/vllm/sampling_params.py

While most of the key parameters are already supported, some of them are missing:

TODO (help wanted):

Parameters that will not be supported in V1:

  • best_of
  • logits_processors

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.
@22quinn
Copy link

22quinn commented Feb 12, 2025

Seems like good first issues - @WoosukKwon I'd like to take a stab at bad_words

@akeshet
Copy link
Contributor

akeshet commented Feb 12, 2025

Parameters that will not be supported in V1:

...
logits_processors

Is this to say that support for logits_processors is going to be permanently dropped in V1, or is it just out of scope for this particular GH issue?

@WoosukKwon
Copy link
Collaborator Author

@22quinn Thanks for volunteering! Could you please submit a PR by EoW?

@WoosukKwon
Copy link
Collaborator Author

@akeshet

We plan to re-design the API for that. We will probably not allow per-request logits processor (because this is too complex and slow). We are exploring other options.

Please refer to this comment from @njhill: #12688 (comment)

@njhill
Copy link
Member

njhill commented Feb 12, 2025

We'll still have a logits processor plugin mechanism (not per request) but the interface will be different. I think that @AlpinDale was working on a proposal for this.

It might be nice for some of these sampling parameters to be implemented via that same abstraction but assuming it won't be ready before we need them we can plan to retrofit where it makes sense.

@maliknaik16
Copy link

@22quinn, are you still working on this task? If not, I'd be happy to take it up! Let me know what you think.

cc @WoosukKwon

@WoosukKwon
Copy link
Collaborator Author

@maliknaik16 Please feel free to take it!

@22quinn Let us know if you already have the PR.

@22quinn
Copy link

22quinn commented Feb 16, 2025

@maliknaik16 @WoosukKwon I'm on it, PR will be up soon

@WoosukKwon
Copy link
Collaborator Author

@22quinn Oh great. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
6 participants