diff --git a/examples/python/model-chat.py b/examples/python/model-chat.py index 421d578a7..f95667a18 100644 --- a/examples/python/model-chat.py +++ b/examples/python/model-chat.py @@ -78,7 +78,7 @@ def main(args): system_prompt = f"<|im_start|>system<|im_sep|>\n{args.system_prompt}<|im_end|>" elif model_type.startswith("llama"): system_prompt = f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n{args.system_prompt}<|eot_id|>" - print("Using System Prompt for LLAMA 3, if you are using LLAMA 2 please pass the argument --system_prompt '[INST] <>\n{args.system_prompt}\n<>')") + print("Using System Prompt for LLAMA 3, if you are using LLAMA 2 please pass the argument --system_prompt '[INST] <>\\n{args.system_prompt}\\n<>')") elif model_type.startswith("qwen2"): system_prompt = f"<|im_start|>system\n{args.system_prompt}<|im_end|>\n" else: diff --git a/examples/python/model-qa.py b/examples/python/model-qa.py index 0eeb2dcf4..f2adf9387 100644 --- a/examples/python/model-qa.py +++ b/examples/python/model-qa.py @@ -66,7 +66,7 @@ def main(args): system_prompt = f"<|im_start|>system<|im_sep|>\n{args.system_prompt}<|im_end|>" elif model_type.startswith("llama"): system_prompt = f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n{args.system_prompt}<|eot_id|>" - print("Using System Prompt for LLAMA 3, if you are using LLAMA 2 please pass the argument --system_prompt '[INST] <>\n{args.system_prompt}\n<>')") + print("Using System Prompt for LLAMA 3, if you are using LLAMA 2 please pass the argument --system_prompt '[INST] <>\\n{args.system_prompt}\\n<>')") elif model_type.startswith("qwen2"): system_prompt = f"<|im_start|>system\n{args.system_prompt}<|im_end|>\n" else: