From 1ba78207e6457ad2f7eba6ca145465bdd6ca2adf Mon Sep 17 00:00:00 2001
From: Xin Yang <105740670+xyang16@users.noreply.github.com>
Date: Mon, 22 May 2023 01:10:21 -0700
Subject: [PATCH] fix: set MODEL_ID in sagemaker-entrypoint script (#343)

---
 sagemaker-entrypoint.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/sagemaker-entrypoint.sh b/sagemaker-entrypoint.sh
index 711e3721b..1340f55a7 100755
--- 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}"
@@ -17,4 +18,4 @@ if [[ -n "${HF_MODEL_QUANTIZE}" ]]; then
   export QUANTIZE="${HF_MODEL_QUANTIZE}"
 fi
 
-text-generation-launcher --port 8080
\ No newline at end of file
+text-generation-launcher --port 8080