From e327044acc48c8e5ac18e718a7d3885e9c4ab3e9 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 1 Dec 2024 02:20:54 -0800 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 68aa527990..8f2e56d6e6 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1736,11 +1736,11 @@ class FastLlamaModel: print(debug_info) debug_info = debug_info\ - .replace("{unsloth_0}", unsloth_0)\ - .replace("{unsloth_1}", unsloth_1)\ - .replace("{unsloth_2}", unsloth_2)\ - .replace("{unsloth_3}", unsloth_3)\ - .replace("{unsloth_4}", unsloth_4) + .replace("{unsloth_0}", re.escape(unsloth_0))\ + .replace("{unsloth_1}", re.escape(unsloth_1))\ + .replace("{unsloth_2}", re.escape(unsloth_2))\ + .replace("{unsloth_3}", re.escape(unsloth_3))\ + .replace("{unsloth_4}", re.escape(unsloth_4)) debug_info = debug_info.split('\n') debug_info = "\n".join([debug_info[0]] + [spaces + x[8:] for x in debug_info[1:]])