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

madamczykhabana · Jun 2, 2023 · 62fc401 · 62fc401
1 parent e7248fe
commit 62fc401
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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