From 62fc40103079bc27e97194ef69e9e34a180b0a85 Mon Sep 17 00:00:00 2001 From: OlivierDehaene Date: Fri, 2 Jun 2023 09:51:06 +0200 Subject: [PATCH] feat(sagemaker): add trust remote code to entrypoint (#394) --- sagemaker-entrypoint.sh | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/sagemaker-entrypoint.sh b/sagemaker-entrypoint.sh index 1340f55..9ac4701 100755 --- 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