Update llama.py

unsloth/models/llama.py

@@ -1717,7 +1717,7 @@ class FastLlamaModel:
         f"{unsloth_2}  Batch size per device = {self._train_batch_size:,} | Gradient Accumulation steps = {args.gradient_accumulation_steps}\\n"\\
         f"{unsloth_3}  Total batch size = {total_train_batch_size:,} | Total steps = {max_steps:,}\\n"\\
         f'{unsloth_4}  Number of trainable parameters = {get_model_param_count(model, trainable_only=True):,}'
-        logger.warning(debug_info)
+        print(debug_info)
         import subprocess, re, gc, numpy as np
         a = np.array([0,])
         try: