diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index 1f6a589ccc14..dfbc7a7b302e 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -1,6 +1,6 @@ [dev] # Set to "huggingface", for example, if you are a huggingface developer. Default is "" -partner_developer = "" +partner_developer = "huggingface" # Please only set it to true if you are preparing an EI related PR # Do remember to revert it back to false before merging any PR (including EI dedicated PR) ei_mode = false @@ -9,7 +9,7 @@ ei_mode = false neuron_mode = false # Please only set it to true if you are preparing a NEURONX related PR # Do remember to revert it back to false before merging any PR (including NEURONX dedicated PR) -neuronx_mode = false +neuronx_mode = true # Please only set it to true if you are preparing a GRAVITON related PR # Do remember to revert it back to false before merging any PR (including GRAVITON dedicated PR) graviton_mode = false @@ -34,11 +34,11 @@ 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 - ["autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "mxnet", "pytorch", "stabilityai_pytorch"] -build_frameworks = [] +build_frameworks = ["huggingface_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"