-
Notifications
You must be signed in to change notification settings - Fork 18
✅ add tokenizer option for testing #248
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
Conversation
Signed-off-by: Prashant Gupta <[email protected]>
👋 Hi! Thank you for contributing to vLLM support on Spyre.
Or this can be done with
Now you are good to go 🚀 |
@@ -44,6 +45,8 @@ | |||
], | |||
ids=lambda val: f"num_prompts({len(val)})", | |||
) | |||
@pytest.mark.parametrize("max_num_seqs", [2, 3, 4], | |||
ids=lambda val: f"max_num_seqs({val})") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We want this to be changed less frequently while parameterizing tests
Signed-off-by: Prashant Gupta <[email protected]>
Signed-off-by: Prashant Gupta <[email protected]>
Signed-off-by: Prashant Gupta <[email protected]>
Signed-off-by: Prashant Gupta <[email protected]>
Signed-off-by: Prashant Gupta <[email protected]>
Signed-off-by: Prashant Gupta <[email protected]>
"9 8 7 6", | ||
], | ||
[ | ||
"7 6 5 4", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
added prompt
with len=1
for testing
Signed-off-by: Prashant Gupta <[email protected]>
Description
Add
VLLM_SPYRE_TEST_TOKENIZER_LIST
as an optional parameter while testing. This is used bytiny-granite
models which need to supplytokenizer
as an additional input to vllm.then
will start with
Also changes
max_model_len
to256
in CB tests.