How to pass parameter in ensemble model?

[GooVincent]

As the normal procedure for tensorrtllm_backend is preprocessing -> (tensorrt_llm) process -> postprocessing. How to pass the customer parameter from the request, like request token length.

In my understanding, tensorrt_llm backend will finish the infer, it won't work to add input and output parameter. then the issue coming, in ensemble pipeline, how to pass the parameter from the preprocess module to poseprocess module.

Please any way to solve this issue?

[byshiue]

I think I don't really get your question. max_tokens (here) is a input form parameter, which is similar to request token length. Can you explain more for your question?

[GooVincent]

Well, I understood how to pass the parameter from the model preprocess to model tensorrt_llm, like REQUEST_INPUT_LEN

but how to pass REQUEST_INPUT_LEN from model tensorrt_llm to the model postprocess, I get confused, as the output for tensorrt_llm should be fixed based the llm model. correct?

[byshiue]

For ensemble, you should pass REQUEST_INPUT_LEN for ensemble to postprocess, instead of passing from tensorrt_llm to postprocess.

[GooVincent]

Any guide for this please? as the REQUEST_INPUT_LEN is the intermediate result.

[byshiue]

I get your point. As far as I know, there is no way to support such feature in tensorrt_llm backend directly because it needs to change the source code of batch_manager. You can try to map the output of preporcess to input of postprocess directly. I am not sure is it doable. You can ask in tritonserver repo.

[GooVincent]

okay, thanks.

[GooVincent]

I get your point. As far as I know, there is no way to support such feature in tensorrt_llm backend directly because it needs to change the source code of batch_manager. You can try to map the output of preporcess to input of postprocess directly. I am not sure is it doable. You can ask in tritonserver repo.

Any way for the ensemble mode to know the request token length, as I wanna cut the original tokens.

[BasicCoder]

I get your point. As far as I know, there is no way to support such feature in tensorrt_llm backend directly because it needs to change the source code of batch_manager. You can try to map the output of preporcess to input of postprocess directly. I am not sure is it doable. You can ask in tritonserver repo.

Any way for the ensemble mode to know the request token length, as I wanna cut the original tokens.

May be you can reference this #95

[callmezhangchenchenokay]

@GooVincent Hi ，
Have you solved it yet

[callmezhangchenchenokay]

Here is a demo, you can pass out the REQUEST_INPUT_LEN in the middle,
I've tested it and it's okay

https://github.com/triton-inference-server/tensorrtllm_backend/blob/57de9f572f75f61fe17b668eea1430b030e1b721/all_models/inflight_batcher_llm/postprocessing/config.pbtxt

[byshiue]

Since the issue is solved, close this bug. If you still have issue/question, feel free to ask and we will open the issue again.