Update loader.py

unslothai · Dec 29, 2024 · ea2c647 · ea2c647
1 parent 5495311
commit ea2c647
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py
@@ -454,7 +454,7 @@ def from_pretrained(
 
         if not was_disabled: enable_progress_bars()
 
-        if True: #with contextlib.redirect_stdout(open(os.devnull, "w")):
+        with contextlib.redirect_stdout(open(os.devnull, "w")):
             patch_loss_functions(torch_compile = False)
             model_types = unsloth_compile_transformers(
                 model_name              = model_name,
@@ -470,7 +470,7 @@ def from_pretrained(
                 fuse_lm_head            = True,
                 gradient_checkpointing  = True,
                 manual_replacements     = True,
-                fast_lora_forwards      = True,
+                fast_lora_forwards      = False,
                 fast_residual_stream    = False,
                 accurate_accumulation   = True,
                 epilogue_fusion         = True,