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)