fix: set MODEL_ID in sagemaker-entrypoint script (#343)

2023-05-22 01:10:21 -07:00 · 2023-05-22 01:10:21 -07:00 · 1ba78207e6
parent 5a58226130
commit 1ba78207e6
1 changed files with 2 additions and 1 deletions
--- a/sagemaker-entrypoint.sh
+++ b/sagemaker-entrypoint.sh
@ -4,6 +4,7 @@ if [[ -z "${HF_MODEL_ID}" ]]; then
  echo "HF_MODEL_ID must be set"
  exit 1
 fi
+export MODEL_ID="${HF_MODEL_ID}"

 if [[ -n "${HF_MODEL_REVISION}" ]]; then
  export REVISION="${HF_MODEL_REVISION}"