Set MODEL_ID in sagemaker-entrypoint script

huggingface · May 19, 2023 · deded7d · deded7d
1 parent 5a58226
commit deded7d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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
+text-generation-launcher --port 8080