From 6d468532e127aa096877a062ea68eb2f55445386 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 1 Dec 2024 02:13:34 -0800 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 8c4c5e6f68..43d0e9f1c7 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1706,13 +1706,12 @@ class FastLlamaModel: spaces = re.search('\n([\s\t]{1,})', original_debug).group(0)[1:] front_spaces = re.match('([\s\t]{1,})', inner_training_loop).group(0) + debug_info = """debug_info = \\ unsloth_0 = r'==((====))==' unsloth_1 = r' \\ /| ' unsloth_2 = r'O^O/ \_/ \ ' unsloth_3 = r'\ / ' unsloth_4 = r' "-____-" ' - - debug_info = """debug_info = \\ f"{unsloth_0} Unsloth - 2x faster free finetuning | Num GPUs = {args.world_size}\\n"\\ f"{unsloth_1} Num examples = {num_examples:,} | Num Epochs = {num_train_epochs:,}\\n"\\ f"{unsloth_2} Batch size per device = {self._train_batch_size:,} | Gradient Accumulation steps = {args.gradient_accumulation_steps}\\n"\\