diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index b01cb4113b..4a57de3aba 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -280,6 +280,17 @@ if os.environ.get("UNSLOTH_ENABLE_LOGGING", "0") != "1": del vllm_executor_logger except: pass + try: + from vllm.v1.executor.abstract import logger as vllm_v1_executor_logger + + vllm_v1_executor_logger.addFilter(HideLoggingMessage("to fall asleep")) + vllm_v1_executor_logger.addFilter(HideLoggingMessage("to wake up")) + vllm_v1_executor_logger.addFilter( + HideLoggingMessage("Executor is not sleeping") + ) + del vllm_v1_executor_logger + except: + pass try: from vllm.core.block.prefix_caching_block import ( logger as vllm_prefix_caching_logger,