Skip to content

Commit 4ef979a

Browse files
author
malavhs
committed
revert dlc_developer_config.toml
1 parent 9742b95 commit 4ef979a

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

dlc_developer_config.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,10 @@ deep_canary_mode = false
3737
[build]
3838
# Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
3939
# available frameworks - ["autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "mxnet", "pytorch", "stabilityai_pytorch"]
40-
build_frameworks = ["huggingface_pytorch"]
40+
build_frameworks = []
4141

4242
# By default we build both training and inference containers. Set true/false values to determine which to build.
43-
build_training = false
43+
build_training = true
4444
build_inference = true
4545

4646
# Set do_build to "false" to skip builds and test the latest image built by this PR
@@ -146,7 +146,7 @@ dlc-pr-tensorflow-2-neuron-inference = ""
146146

147147
# HuggingFace Inference
148148
dlc-pr-huggingface-tensorflow-inference = ""
149-
dlc-pr-huggingface-pytorch-inference = "huggingface/pytorch/inference/buildspec.yml"
149+
dlc-pr-huggingface-pytorch-inference = ""
150150
dlc-pr-huggingface-pytorch-neuron-inference = ""
151151

152152
# Stability AI Inference

test/sagemaker_tests/huggingface/inference/integration/local/test_serving.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def _assert_prediction(predictor):
6161
@pytest.mark.model("tiny-distilbert")
6262
@pytest.mark.team("sagemaker-1p-algorithms")
6363
def test_serve_json(docker_image, framework_version, sagemaker_local_session, instance_type):
64-
if "pytorch" in docker_image and Version(framework_version) < Version("2.4"):
64+
if "huggingface-pytorch" in docker_image and Version(framework_version) < Version("2.4"):
6565
pytest.skip("Skipping distilbert SM local tests for PT")
6666
with _predictor(
6767
model_dir, docker_image, framework_version, sagemaker_local_session, instance_type

0 commit comments

Comments
 (0)