From 2ddd8d285bb7f0afc02eebc17edade921005b4e5 Mon Sep 17 00:00:00 2001 From: Ethan Zhang Date: Thu, 13 Jun 2024 23:21:28 +0000 Subject: [PATCH] [secure-mode] Fix entrypoint control name --- .../java/ai/djl/serving/plugins/securemode/SecureModeUtils.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/plugins/secure-mode/src/main/java/ai/djl/serving/plugins/securemode/SecureModeUtils.java b/plugins/secure-mode/src/main/java/ai/djl/serving/plugins/securemode/SecureModeUtils.java index e36fceda4..e5278e715 100644 --- a/plugins/secure-mode/src/main/java/ai/djl/serving/plugins/securemode/SecureModeUtils.java +++ b/plugins/secure-mode/src/main/java/ai/djl/serving/plugins/securemode/SecureModeUtils.java @@ -48,7 +48,7 @@ final class SecureModeUtils { static final String REQUIREMENTS_TXT_CONTROL = "DISALLOW_REQUIREMENTS_TXT"; static final String PICKLE_FILES_CONTROL = "DISALLOW_PICKLE_FILES"; static final String TRUST_REMOTE_CODE_CONTROL = "DISALLOW_TRUST_REMOTE_CODE"; - static final String CUSTOM_ENTRYPOINT_CONTROL = "DISALLOW_CUSTOM_ENTRYPOINT"; + static final String CUSTOM_ENTRYPOINT_CONTROL = "DISALLOW_CUSTOM_INFERENCE_SCRIPTS"; static final String CHAT_TEMPLATE_CONTROL = "DISALLOW_CHAT_TEMPLATE"; private static final Pattern PICKLE_EXTENSIONS_REGEX =