11 lines
254 B
Bash
11 lines
254 B
Bash
|
#!/bin/bash
|
||
|
|
set -ex
|
||
|
|
|
||
|
|
if [ -n "${S3_MODEL_DIR}" ]; then
|
||
|
|
aws s3 sync --exclude '*/dataset/*' --exclude '*/cache/*' --exclude 'iteration_*.pth' --exclude '*_optim.pth' "${S3_MODEL_DIR}" ./model
|
||
|
|
genienlp kfserver --path ./model $@
|
||
|
|
else
|
||
|
|
genienlp $@
|
||
|
|
fi
|
||
|
|
|