Skip to content

Commit d7ac566

Browse files
committed
run 2.6 inf on autopatch
1 parent cce532f commit d7ac566

File tree

2 files changed

+7
-10
lines changed

2 files changed

+7
-10
lines changed

dlc_developer_config.toml

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -36,17 +36,17 @@ deep_canary_mode = false
3636

3737
[build]
3838
# Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
39-
# available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_vllm", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
40-
build_frameworks = []
39+
# available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
40+
build_frameworks = ["pytorch"]
4141

4242

4343
# By default we build both training and inference containers. Set true/false values to determine which to build.
44-
build_training = true
45-
build_inference = true
44+
build_training = false
45+
build_inference = true
4646

4747
# Set do_build to "false" to skip builds and test the latest image built by this PR
4848
# Note: at least one build is required to set do_build to "false"
49-
do_build = true
49+
do_build = true
5050

5151
[notify]
5252
### Notify on test failures
@@ -154,7 +154,7 @@ dlc-pr-tensorflow-2-habana-training = ""
154154
### INFERENCE PR JOBS ###
155155

156156
# Standard Framework Inference
157-
dlc-pr-pytorch-inference = ""
157+
dlc-pr-pytorch-inference = "pytorch/inference/buildspec-2-6-sm.yml"
158158
dlc-pr-tensorflow-2-inference = ""
159159
dlc-pr-autogluon-inference = ""
160160

@@ -186,8 +186,5 @@ dlc-pr-tensorflow-2-eia-inference = ""
186186
# vllm
187187
dlc-pr-vllm = ""
188188

189-
# HuggingFace vLLM
190-
dlc-pr-huggingface-vllm = ""
191-
192189
# sglang
193190
dlc-pr-sglang = ""

pytorch/inference/buildspec-2-6-sm.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
55
version: &VERSION 2.6.0
66
short_version: &SHORT_VERSION "2.6"
77
arch_type: x86
8-
# autopatch_build: "True"
8+
autopatch_build: "True"
99

1010
repository_info:
1111
inference_repository: &INFERENCE_REPOSITORY

0 commit comments

Comments
 (0)