#!/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