From a356b0e228b640a31cd9d1718d097a1aabcd7675 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=2E=20Yusuf=20Sar=C4=B1g=C3=B6z?= <yusufsarigoz@gmail.com>
Date: Fri, 11 Aug 2023 10:50:02 +0300
Subject: [PATCH] gguf : start implementing gguf_file_saver (WIP)

---
 gguf-llama.cpp | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/gguf-llama.cpp b/gguf-llama.cpp
index 9c0b5651a..e88dc6a08 100644
--- a/gguf-llama.cpp
+++ b/gguf-llama.cpp
@@ -625,20 +625,25 @@ struct gguf_file_saver {
     gguf_file_saver(const char * fname, gguf_file_loader * any_file_loader, enum llama_ftype new_ftype)
         : file(fname, "wb"), any_file_loader(any_file_loader) {
         fprintf(stderr, "llama.cpp: saving model to %s\n", fname);
-        write_magic();
-        write_version();
+        write_header();
         write_hparams(new_ftype);
         write_vocab();
     }
 
-    void write_magic() {
+    // TODO: probably it's better to move these to gguf_file
+
+    void write_header() {
         const int32_t magic = GGUF_MAGIC;
         file.write_i32(magic);
-        }
-
-        void write_version() {
+    
             const int32_t version = GGUF_VERSION;
             file.write_i32(version);
+
+            const int32_t n_tensors = gguf_get_n_tensors(any_file_loader->gguf_ctx);
+            file.write_i32(n_tensors);
+
+            const int32_t n_kv = gguf_get_n_kv(any_file_loader->gguf_ctx);
+            file.write_i32(n_kv);
         }
 
     void write_hparams(enum llama_ftype new_ftype) {
@@ -651,7 +656,7 @@ struct gguf_file_saver {
         uint32_t n_vocab = any_file_loader->hparams.n_vocab;
         GGML_UNUSED(n_vocab);
     }
-    
+
     void write_tensor(llama_load_tensor & tensor, enum ggml_type new_type, const void * new_data, size_t new_size) {
         switch (new_type) {
             case GGML_TYPE_F32: