Skip to content

Commit ee050e4

Browse files
authored
fix: CPU OOM issue during LoRA training (#41)
1 parent 7803632 commit ee050e4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

mova/engine/trainer/accelerate/accelerate_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -276,7 +276,7 @@ def _setup(self):
276276
if self.is_main_process:
277277
os.makedirs(self.save_path, exist_ok=True)
278278

279-
torch.cuda.memory._record_memory_history(enabled='all')
279+
# torch.cuda.memory._record_memory_history(enabled='all')
280280

281281
def _setup_lora(self):
282282
"""Configure LoRA"""

0 commit comments

Comments
 (0)