From e798d822f57e34550f1f4f6f2040c1e14641d367 Mon Sep 17 00:00:00 2001
From: Wellington Marthas <w@dyntime.com>
Date: Sun, 22 Feb 2026 04:47:50 -0300
Subject: [PATCH] Fix resume argument in entrypoint

---
 entrypoint.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/entrypoint.sh b/entrypoint.sh
index 7deec3f..1f1dce4 100644
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -7,7 +7,7 @@ CKPT=""
 
 if compgen -G "${CKPT_DIR}/version_*/checkpoints/*.ckpt" > /dev/null; then
   CKPT=$(ls -t ${CKPT_DIR}/version_*/checkpoints/*.ckpt | head -n1)
-  RESUME_ARG="--trainer.resume_from_checkpoint $CKPT"
+  RESUME_ARG="--ckpt_path $CKPT"
 
   echo "Resuming: $CKPT"
 fi