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

Add skip-pull flag to allow local build #5186

Merged
merged 3 commits into from
Dec 21, 2022
Merged

Conversation

jbkyang-nvi
Copy link
Contributor

@jbkyang-nvi jbkyang-nvi commented Dec 20, 2022

Triton has backends that are not shipped with the official containers such as
FasterTransformer Backend.
This PR allows compose.py to use local images so users can further decrease the size of the images
with these more customized backends.

Use case:
python3 compose.py --backend pytorch --backend fastertransformer --image full,triton_with_ft:22.11 --image gpu-min,nvcr.io/nvidia/tritonserver:22.11-py3-min --image min,nvcr.io/nvidia/tritonserver:22.11-py3-min --skip-pull

compose.py Outdated Show resolved Hide resolved
compose.py Outdated Show resolved Hide resolved
compose.py Outdated Show resolved Hide resolved
GuanLuo
GuanLuo previously approved these changes Dec 21, 2022
@jbkyang-nvi jbkyang-nvi merged commit db024ab into main Dec 21, 2022
@jbkyang-nvi jbkyang-nvi deleted the kyang-add-compose-local-flag branch December 21, 2022 22:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Development

Successfully merging this pull request may close these issues.

3 participants