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 =