a | b/docker/serve/entrypoint.sh | ||
---|---|---|---|
1 | #!/bin/bash |
||
2 | set -e |
||
3 | |||
4 | if [[ "$1" = "serve" ]]; then |
||
5 | shift 1 |
||
6 | torchserve --start --ts-config /home/model-server/config.properties |
||
7 | else |
||
8 | eval "$@" |
||
9 | fi |
||
10 | |||
11 | # prevent docker exit |
||
12 | tail -f /dev/null |