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

How to configure max_num_tokens and max_batch_size as runtime params? #2490

Open
JoJoLev opened this issue Nov 24, 2024 · 2 comments
Open

How to configure max_num_tokens and max_batch_size as runtime params? #2490

JoJoLev opened this issue Nov 24, 2024 · 2 comments
Assignees
Labels
question Further information is requested Triton Backend

Comments

@JoJoLev
Copy link

JoJoLev commented Nov 24, 2024

I noticed that these values can be configured at runtime, does this mean I must define them in the config.pbtxt if running on Triton?
Or should I pass the values as env variables of some sort since I am running this on Sagemaker?

Thanks!

@hello-11 hello-11 added Triton Backend question Further information is requested labels Nov 25, 2024
@nv-guomingz
Copy link
Collaborator

Hi @Tabrizian would u please take a look this question?

@Tabrizian
Copy link
Collaborator

@JoJoLev For runtime values that are supported in model configuration you need to specify them in Triton's config.pbtxt file. There is documentation for the parameters that are supported here:

https://github.com/triton-inference-server/tensorrtllm_backend/blob/main/docs/model_config.md#common-inputs

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested Triton Backend
Projects
None yet
Development

No branches or pull requests

4 participants