From a7f984e22ecab9cbb575ffdf988ae99facef3ea2 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 18 Mar 2025 01:05:58 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 3b4e276bbd..e2b35c5ff6 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -182,6 +182,15 @@ try: except: pass +# Gemma3 It is strongly recommended to train Gemma3 models with the `eager` +try: + from transformers.models.gemma3.modeling_gemma3 import logger as gemma3_logger + gemma3_logger.addFilter(HideLoggingMessage("strongly recommended")) + del gemma3_logger +except: + pass + + # Patch get_model_param_count to record correct 4bit / 8bit from transformers.trainer_pt_utils import is_deepspeed_zero3_enabled def get_model_param_count(model, trainable_only = False):