diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index 207784d1f191..1e6f4e1ad2d0 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -37,12 +37,12 @@ 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 = [] +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_inference = false # 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" @@ -124,7 +124,7 @@ nightly_pr_test_mode = false dlc-pr-base = "" # Standard Framework Training -dlc-pr-pytorch-training = "" +dlc-pr-pytorch-training = "pytorch/training/buildspec-2-9-sm.yml" dlc-pr-tensorflow-2-training = "" dlc-pr-autogluon-training = "" diff --git a/pytorch/training/buildspec-2-9-ec2.yml b/pytorch/training/buildspec-2-9-ec2.yml index 3f2cef8d599f..555ea9d0c2db 100644 --- a/pytorch/training/buildspec-2-9-ec2.yml +++ b/pytorch/training/buildspec-2-9-ec2.yml @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch version: &VERSION 2.9.0 short_version: &SHORT_VERSION "2.9" arch_type: x86 -# autopatch_build: "True" +autopatch_build: "True" repository_info: training_repository: &TRAINING_REPOSITORY diff --git a/pytorch/training/buildspec-2-9-sm.yml b/pytorch/training/buildspec-2-9-sm.yml index c4b2f8dcdff2..5df4edec0b7d 100644 --- a/pytorch/training/buildspec-2-9-sm.yml +++ b/pytorch/training/buildspec-2-9-sm.yml @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch version: &VERSION 2.9.0 short_version: &SHORT_VERSION "2.9" arch_type: x86 -# autopatch_build: "True" +autopatch_build: "True" repository_info: training_repository: &TRAINING_REPOSITORY