feat(sagemaker): add trust remote code to entrypoint (#394)

2023-06-02 09:51:06 +02:00 · 2023-06-02 09:51:06 +02:00 · 62fc401030
parent e7248fe90e
commit 62fc401030
1 changed files with 4 additions and 0 deletions
--- a/sagemaker-entrypoint.sh
+++ b/sagemaker-entrypoint.sh
@ -18,4 +18,8 @@ if [[ -n "${HF_MODEL_QUANTIZE}" ]]; then
  export QUANTIZE="${HF_MODEL_QUANTIZE}"
 fi

+if [[ -n "${HF_MODEL_TRUST_REMOTE_CODE}" ]]; then
+  export TRUST_REMOTE_CODE="${HF_MODEL_TRUST_REMOTE_CODE}"
+fi
+
 text-generation-launcher --port 8080