From a047145caf07f96590484cab639501311234c4bd Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 8 Mar 2025 03:35:22 -0800 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index b4639f27b1..de25ec0b51 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -518,9 +518,12 @@ class FastModel(FastBaseModel): if not was_disabled: enable_progress_bars() do_logging = os.environ.get("UNSLOTH_ENABLE_LOGGING", "0") == "1" - redirector = sys.stdout if do_logging else open(os.devnull, "w") + if do_logging: + redirector = contextlib.redirect_stdout(open(os.devnull, "w")) + else: + redirector = contextlib.nullcontext() - with contextlib.redirect_stdout(redirector): + with redirector: patch_loss_functions(torch_compile = False) model_types = unsloth_compile_transformers( model_name = model_name,