diff --git a/examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py b/examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py
index 55a666774..3313ac4ca 100644
--- a/examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py
+++ b/examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py
@@ -412,6 +412,8 @@ class Checkpoint:
         return offset
 
     def save_gguf(self, gguf_writer):
+        gguf_writer.add_architecture()
+        gguf_writer.add_file_type(gguf.GGMLQuantizationType.F32)
         gguf_writer.add_uint32(LLM_KV_TRAINING_FILE_VERSION,    0)
         gguf_writer.add_uint32(LLM_KV_TRAINING_ITERATION_COUNT, self.train_its)
         gguf_writer.add_uint32(LLM_KV_TRAINING_SAMPLE_COUNT,    self.train_samples)