diff --git a/modules/chat.py b/modules/chat.py index 348c5eba..bddc3132 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -208,7 +208,7 @@ def generate_chat_prompt(user_input, state, **kwargs): logger.error(f"Failed to build the chat prompt. The input is too long for the available context length.\n\nTruncation length: {state['truncation_length']}\nmax_new_tokens: {state['max_new_tokens']} (is it too high?)\nAvailable context length: {max_length}\n") raise ValueError else: - logger.warning(f"The input has been truncated. Context length: {state['truncation_length']}, max_new_tokens: {state['max_new_tokens']}.") + logger.warning(f"The input has been truncated. Context length: {state['truncation_length']}, max_new_tokens: {state['max_new_tokens']}, available context length: {max_length}.") break prompt = make_prompt(messages)