From 71d215f0bd99b8223af481f99f989d5ba1b0ed2c Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Thu, 1 Jun 2023 11:00:05 +0200
Subject: [PATCH] feat(sagemaker): add trust remote code to entrypoint

---
 sagemaker-entrypoint.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/sagemaker-entrypoint.sh b/sagemaker-entrypoint.sh
index 1340f55a..9ac47010 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