Skip to content

Commit

Permalink
Update lora_dpo_single_device.py
Browse files Browse the repository at this point in the history
  • Loading branch information
krammnic authored Oct 24, 2024
1 parent 641d081 commit 42b51fd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion recipes/lora_dpo_single_device.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ def __init__(self, cfg: DictConfig) -> None:

if self._log_peak_memory_stats and self._device.type == "cuda":
log.info(
"log_peak_memory_stats was se to True, however, training does not use cuda. Setting log_peak_memory_stats=False."
"log_peak_memory_stats was set to True, however, training does not use cuda. Setting log_peak_memory_stats=False."
)
self._log_peak_memory_stats = False

Expand Down

0 comments on commit 42b51fd

Please sign in to comment.