diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index 207784d1f191..febfffe6fd5f 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -36,17 +36,17 @@ deep_canary_mode = false [build] # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image. -# available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_vllm", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"] -build_frameworks = [] +# available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"] +build_frameworks = ["pytorch"] # By default we build both training and inference containers. Set true/false values to determine which to build. -build_training = true -build_inference = true +build_training = false +build_inference = true # Set do_build to "false" to skip builds and test the latest image built by this PR # Note: at least one build is required to set do_build to "false" -do_build = true +do_build = true [notify] ### Notify on test failures @@ -154,7 +154,7 @@ dlc-pr-tensorflow-2-habana-training = "" ### INFERENCE PR JOBS ### # Standard Framework Inference -dlc-pr-pytorch-inference = "" +dlc-pr-pytorch-inference = "pytorch/inference/buildspec-2-6-ec2.yml" dlc-pr-tensorflow-2-inference = "" dlc-pr-autogluon-inference = "" @@ -186,8 +186,5 @@ dlc-pr-tensorflow-2-eia-inference = "" # vllm dlc-pr-vllm = "" -# HuggingFace vLLM -dlc-pr-huggingface-vllm = "" - # sglang dlc-pr-sglang = "" \ No newline at end of file diff --git a/pytorch/inference/buildspec-2-6-ec2.yml b/pytorch/inference/buildspec-2-6-ec2.yml index 3c75610bde0e..6eab2fbf607d 100644 --- a/pytorch/inference/buildspec-2-6-ec2.yml +++ b/pytorch/inference/buildspec-2-6-ec2.yml @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch version: &VERSION 2.6.0 short_version: &SHORT_VERSION "2.6" arch_type: x86 -# autopatch_build: "True" +autopatch_build: "True" repository_info: inference_repository: &INFERENCE_REPOSITORY diff --git a/pytorch/inference/buildspec-2-6-sm.yml b/pytorch/inference/buildspec-2-6-sm.yml index ab2d6ec6947a..16c2c731631e 100644 --- a/pytorch/inference/buildspec-2-6-sm.yml +++ b/pytorch/inference/buildspec-2-6-sm.yml @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch version: &VERSION 2.6.0 short_version: &SHORT_VERSION "2.6" arch_type: x86 -# autopatch_build: "True" +autopatch_build: "True" repository_info: inference_repository: &INFERENCE_REPOSITORY