From 065167930295ab2eb2974ad53407c24d6426545e Mon Sep 17 00:00:00 2001
From: xaedes <xaedes@gmail.com>
Date: Mon, 22 May 2023 16:56:28 +0200
Subject: [PATCH] save checkpoint only when it was trained

---
 examples/baby-llama/baby-llama-text.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/examples/baby-llama/baby-llama-text.cpp b/examples/baby-llama/baby-llama-text.cpp
index a21403a77..84ef911f8 100644
--- a/examples/baby-llama/baby-llama-text.cpp
+++ b/examples/baby-llama/baby-llama-text.cpp
@@ -2089,7 +2089,9 @@ int main(int argc, char ** argv) {
         ggml_free(ctx0);
     }
 
-    save_checkpoint(&model, opt, fn_chkpt_out);
+    if (n_examples > 0) {
+        save_checkpoint(&model, opt, fn_chkpt_out);
+    }
 
     {
         int n_gen = 1024;