10000 Fixing docker CI by udaij12 · Pull Request #3194 · pytorch/serve · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Fixing docker CI #3194

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Jun 14, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .github/workflows/d 10000 ocker-ci.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ jobs:
working-directory: docker
run: |
IMAGE_TAG=test-image-${{ matrix.python-version }}
./build_image.sh -py "${{ matrix.python-version }}" -t "${IMAGE_TAG}"
./build_image.sh -py "${{ matrix.python-version }}" -t "${IMAGE_TAG}" -s -r
echo "IMAGE_TAG=${IMAGE_TAG}" >> $GITHUB_OUTPUT

- name: Container Healthcheck
Expand All @@ -40,7 +40,7 @@ jobs:
working-directory: docker
run: ./test_container_python_version.sh ${{ steps.image_build.outputs.IMAGE_TAG }} ${{ matrix.python-version }}

- name: Test model running in container with sample image data
- name: Test model running in container with sample image data
working-directory: docker
run: |
./test_container_model_prediction.sh ${{ steps.image_build.outputs.IMAGE_TAG }}
2 changes: 1 addition & 1 deletion docker/build_image.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ do
echo "-py, --pythonversion specify to python version to use: Possible values: 3.8 3.9 3.10"
echo "-n, --nightly specify to build with TorchServe nightly"
echo "-s, --source specify to build with TorchServe from source"
echo "-l, --local specify to use local TorchServe"
echo "-r, --remote specify to use local TorchServe"
exit 0
;;
-b|--branch_name)
Expand Down
2 changes: 1 addition & 1 deletion docker/dockerd-entrypoint.sh
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ set -e

if [[ "$1" = "serve" ]]; then
shift 1
torchserve --start --ts-config /home/model-server/config.properties
torchserve --start --ts-config /home/model-server/config.properties --disable-token
else
eval "$@"
fi
8000 Expand Down
2 changes: 1 addition & 1 deletion docker/test_container_model_prediction.sh
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ torch-model-archiver \
--handler=/home/model-server/mnist_handler.py \
--export-path=/home/model-server/model-store

torchserve --start --ts-config=/home/model-server/config.properties --models mnist=mnist.mar
torchserve --start --ts-config=/home/model-server/config.properties --models mnist=mnist.mar --disable-token
EOF

echo "Starting container ${CONTAINER}"
Expand Down
Loading
0