diff --git a/unsloth/save.py b/unsloth/save.py
index 7f8cfe8ad2..30eaac92ff 100644
--- a/unsloth/save.py
+++ b/unsloth/save.py
@@ -379,7 +379,7 @@ def unsloth_save_model(
         for item in LLAMA_WEIGHTS:
             proj = eval(f"layer.{item}")
             name = f"model.layers.{j}.{item}.weight"
-            W, out_of_memory = _merge_lora(proj, name)
+            W, out_of_memory = _merge_lora(proj, name, max_vram)
 
             if not out_of_memory:
                 # Save to GPU memory