From 4620e76e3dd428b9f99ff6d0733dbfeb60c8a150 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 1 Dec 2024 02:22:21 -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 8f2e56d6e6..67a9d37bc3 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}", 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)) + .replace("{unsloth_0}", re.unescape(unsloth_0))\ + .replace("{unsloth_1}", re.unescape(unsloth_1))\ + .replace("{unsloth_2}", re.unescape(unsloth_2))\ + .replace("{unsloth_3}", re.unescape(unsloth_3))\ + .replace("{unsloth_4}", re.unescape(unsloth_4)) debug_info = debug_info.split('\n') debug_info = "\n".join([debug_info[0]] + [spaces + x[8:] for x in debug_info[1:]])