From b6056c3db86ce55e6b05f1fc988fdbb02004c359 Mon Sep 17 00:00:00 2001
From: klosax <131523366+klosax@users.noreply.github.com>
Date: Tue, 15 Aug 2023 19:53:44 +0200
Subject: [PATCH] gguf.py : add tensor data layout

---
 gguf.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gguf.py b/gguf.py
index d3b95963a..1b15554f3 100644
--- a/gguf.py
+++ b/gguf.py
@@ -207,6 +207,9 @@ class GGUFWriter:
     def add_author(self, author: str):
         self.add_string(constants.KEY_GENERAL_AUTHOR, author)
 
+    def add_tensor_data_layout(self, layout: str):
+        self.add_string(constants.KEY_LLM_TENSOR_DATA_LAYOUT , layout)
+
     def add_url(self, url: str):
         self.add_string(constants.KEY_GENERAL_URL, url)