diff --git a/train_diffusion.py b/train_diffusion.py
index 5408db1..f5a6168 100644
--- a/train_diffusion.py
+++ b/train_diffusion.py
@@ -208,7 +208,7 @@ for epoch in range(EPOCHS):
             print(f"Epoch {epoch} Step {step} Loss {loss.item():.4f}")
 
 # =========================
-# SAVE LoRA CORRETO
+# SAVE LoRA
 # =========================
 
 OUTPUT_DIR.mkdir(parents=True, exist_ok=True)