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"\\