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

feat: add backend param for triton serving #1039

Merged

Conversation

gujingit
Copy link
Contributor

@gujingit gujingit commented Feb 5, 2024

Feature: support vllm、 tensorrt-llm backend for triton serving.

vllm backend

arena serve triton \
 --backend=vllm \
 --name=triton-vllm \
 --namespace=default \
 --cpu=6 \
 --memory=12Gi \
 --gpus=1 \
 --data=model-pvc:/mnt/pvc/ \
 --model-repository=/mnt/pvc/model_repository

trt-llm

arena serve triton \
 --backend=trt-llm \
 --name=triton-trt-llm \
 --namespace=default \
 --cpu=6 \
 --memory=12Gi \
 --gpus=1 \
 --data=model-pvc:/mnt/pvc/ \
 --model-repository=/mnt/pvc/model_repository

@gujingit
Copy link
Contributor Author

gujingit commented Feb 5, 2024

/assign @Syulin7

@Syulin7
Copy link
Collaborator

Syulin7 commented Feb 18, 2024

@gujingit Thanks! Can you submit a new PR to update the document?

@Syulin7
Copy link
Collaborator

Syulin7 commented Feb 18, 2024

/lgtm
/approve

@google-oss-prow google-oss-prow bot added the lgtm label Feb 18, 2024
Copy link

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: gujingit, Syulin7

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@google-oss-prow google-oss-prow bot merged commit f27a678 into kubeflow:master Feb 18, 2024
4 checks passed
@gujingit gujingit deleted the feature/add_triton_backend branch April 17, 2024 07:18
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants