diff --git a/src/sagemaker_pytorch_serving_container/etc/default-ts.properties b/src/sagemaker_pytorch_serving_container/etc/default-ts.properties index bc0f996a..029f6051 100644 --- a/src/sagemaker_pytorch_serving_container/etc/default-ts.properties +++ b/src/sagemaker_pytorch_serving_container/etc/default-ts.properties @@ -1,4 +1,5 @@ # Based on https://github.com/pytorch/serve/blob/master/docs/configuration.md +vmargs=-XX:-UseContainerSupport enable_envvars_config=true decode_input_request=false load_models=ALL diff --git a/src/sagemaker_pytorch_serving_container/etc/mme-ts.properties b/src/sagemaker_pytorch_serving_container/etc/mme-ts.properties index 37621c1c..1ef6c581 100644 --- a/src/sagemaker_pytorch_serving_container/etc/mme-ts.properties +++ b/src/sagemaker_pytorch_serving_container/etc/mme-ts.properties @@ -1,3 +1,4 @@ +vmargs=-XX:-UseContainerSupport enable_envvars_config=true model_store=/ default_service_handler=$$SAGEMAKER_HANDLER$$ diff --git a/tox.ini b/tox.ini index 0a276d89..fb2fc2ea 100644 --- a/tox.ini +++ b/tox.ini @@ -52,7 +52,7 @@ passenv = AWS_DEFAULT_REGION commands = coverage run --rcfile .coveragerc --source sagemaker_pytorch_serving_container -m pytest {posargs} - {env:IGNORE_COVERAGE:} coverage report --fail-under=90 --include *sagemaker_pytorch_serving_container* + {env:IGNORE_COVERAGE:} coverage report --fail-under=90 deps = coverage