From ad3b2dd7a6463fbd1171d76dd3de8a2522a9db40 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 8 Mar 2025 03:38:22 -0800 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index b368eb9506..800c016cc8 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -519,9 +519,9 @@ class FastModel(FastBaseModel): do_logging = os.environ.get("UNSLOTH_ENABLE_LOGGING", "0") == "1" if do_logging: - redirector = contextlib.redirect_stdout(open(os.devnull, "w")) - else: redirector = contextlib.nullcontext() + else: + redirector = contextlib.redirect_stdout(open(os.devnull, "w")) with redirector: patch_loss_functions(torch_compile = False)