From c967bfffca16650fdeecd76fd324fbb240005a70 Mon Sep 17 00:00:00 2001
From: PandaWu <147988104+SongwuJob@users.noreply.github.com>
Date: Fri, 27 Feb 2026 15:58:13 +0800
Subject: [PATCH] fix: CPU OOM issue during LoRA training

---
 mova/engine/trainer/accelerate/accelerate_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mova/engine/trainer/accelerate/accelerate_trainer.py b/mova/engine/trainer/accelerate/accelerate_trainer.py
index 82ec87a..1f25d95 100644
--- a/mova/engine/trainer/accelerate/accelerate_trainer.py
+++ b/mova/engine/trainer/accelerate/accelerate_trainer.py
@@ -276,7 +276,7 @@ def _setup(self):
         if self.is_main_process:
             os.makedirs(self.save_path, exist_ok=True)
         
-        torch.cuda.memory._record_memory_history(enabled='all')
+        # torch.cuda.memory._record_memory_history(enabled='all')
     
     def _setup_lora(self):
         """Configure LoRA"""