13 lines
197 B
Bash
13 lines
197 B
Bash
|
#!/bin/bash
|
||
|
|
set -e
|
||
|
|
|
||
|
|
if [[ "$1" = "serve" ]]; then
|
||
|
|
shift 1
|
||
|
|
torchserve --start --ts-config /home/model-server/config.properties
|
||
|
|
else
|
||
|
|
eval "$@"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# prevent docker exit
|
||
|
|
tail -f /dev/null
|