From ef8abf4c6e637aa4b658c5ec2d83ff09bc306f59 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 18 Feb 2024 19:25:56 +1100 Subject: [PATCH] apache --- unsloth/models/llama.py | 2 +- unsloth/models/mistral.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index d2b457a9cd..632144afb7 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -984,7 +984,7 @@ class FastLlamaModel: front_spaces = re.match('([\s\t]{1,})', inner_training_loop).group(0) debug_info = """debug_info = \\ - f"==((====))== Unsloth - Free Apache OSS license | Num GPUs = {args.world_size}\\n"\\ + f"==((====))== Unsloth - 2x faster free finetuning | Num GPUs = {args.world_size}\\n"\\ f" \\\\\\ /| Num examples = {num_examples:,} | Num Epochs = {num_train_epochs:,}\\n"\\ f"O^O/ \\_/ \\ Batch size per device = {self._train_batch_size:,} | Gradient Accumulation steps = {args.gradient_accumulation_steps}\\n"\\ f"\\ / Total batch size = {total_train_batch_size:,} | Total steps = {max_steps:,}\\n"\\ diff --git a/unsloth/models/mistral.py b/unsloth/models/mistral.py index b591a1da92..3dcad18201 100644 --- a/unsloth/models/mistral.py +++ b/unsloth/models/mistral.py @@ -389,7 +389,7 @@ class FastMistralModel(FastLlamaModel): front_spaces = re.match('([\s\t]{1,})', inner_training_loop).group(0) debug_info = """debug_info = \\ - f"==((====))== Unsloth - Free Apache OSS license | Num GPUs = {args.world_size}\\n"\\ + f"==((====))== Unsloth - 2x faster free finetuning | Num GPUs = {args.world_size}\\n"\\ f" \\\\\\ /| Num examples = {num_examples:,} | Num Epochs = {num_train_epochs:,}\\n"\\ f"O^O/ \\_/ \\ Batch size per device = {self._train_batch_size:,} | Gradient Accumulation steps = {args.gradient_accumulation_steps}\\n"\\ f"\\ / Total batch size = {total_train_batch_size:,} | Total steps = {max_steps:,}\\n"\\