diff --git a/tensorflow_serving/tools/docker/Dockerfile b/tensorflow_serving/tools/docker/Dockerfile index 48e5fa5ee25..80ebcd1392e 100644 --- a/tensorflow_serving/tools/docker/Dockerfile +++ b/tensorflow_serving/tools/docker/Dockerfile @@ -51,6 +51,7 @@ ENV MODEL_NAME=model # Create a script that runs the model server so we can use environment variables # while also passing in arguments from the docker command line RUN echo '#!/bin/bash \n\n\ +shift \n\ tensorflow_model_server --port=8500 --rest_api_port=8501 \ --model_name=${MODEL_NAME} --model_base_path=${MODEL_BASE_PATH}/${MODEL_NAME} \ "$@"' > /usr/bin/tf_serving_entrypoint.sh \