diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 602f7ee90f..76eefc3c3c 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -183,6 +183,8 @@ except: try: from transformers.generation.utils import logger as transformers_generation_utils_logger transformers_generation_utils_logger.addFilter(HideLoggingMessage("Setting `pad_token_id` to `eos_token_id`")) + # "You have set `compile_config` + transformers_generation_utils_logger.addFilter(HideLoggingMessage("compile_config")) del transformers_generation_utils_logger except: pass