From cb42324d6a88c65ae9ee1b08bf44f8bfb58b402c Mon Sep 17 00:00:00 2001 From: xaedes Date: Sun, 27 Aug 2023 23:20:18 +0200 Subject: [PATCH] add gguf arch and ftype --- .../train-text-from-scratch/convert-train-checkpoint-to-gguf.py | 2 ++ 1 file changed, 2 insertions(+) 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)