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

62fc4010 · OlivierDehaene · GitHub · e7248fe9 · 62fc4010
Unverified Commit 62fc4010 authored Jun 02, 2023 by OlivierDehaene Committed by GitHub Jun 02, 2023
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

sagemaker-entrypoint.sh sagemaker-entrypoint.sh +4 -0

No files found.
--- 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