Skip to content

GitLab

Unverified Commit 1ba78207 authored by Xin Yang's avatar Xin Yang Committed by GitHub
Browse files

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

parent 5a582261
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
sagemaker-entrypoint.sh
View file @ 1ba78207
...@@ -4,6 +4,7 @@ if [[ -z "${HF_MODEL_ID}" ]]; then ...@@ -4,6 +4,7 @@ if [[ -z "${HF_MODEL_ID}" ]]; then
echo "HF_MODEL_ID must be set" echo "HF_MODEL_ID must be set"
exit 1 exit 1
fi fi
export MODEL_ID="${HF_MODEL_ID}"
if [[ -n "${HF_MODEL_REVISION}" ]]; then if [[ -n "${HF_MODEL_REVISION}" ]]; then
export REVISION="${HF_MODEL_REVISION}" export REVISION="${HF_MODEL_REVISION}"
...@@ -17,4 +18,4 @@ if [[ -n "${HF_MODEL_QUANTIZE}" ]]; then ...@@ -17,4 +18,4 @@ if [[ -n "${HF_MODEL_QUANTIZE}" ]]; then
export QUANTIZE="${HF_MODEL_QUANTIZE}" export QUANTIZE="${HF_MODEL_QUANTIZE}"
fi fi
text-generation-launcher --port 8080 text-generation-launcher --port 8080
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment