CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-10-31 08:51:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

gguf.py : add tensor data layout

Browse Source
This commit is contained in:
klosax
2023-08-15 19:53:44 +02:00
committed by GitHub
parent 2dd5d2c92c
commit b6056c3db8
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gguf.py
View File

@@ -207,6 +207,9 @@ class GGUFWriter:
def add_author(self, author: str): def add_author(self, author: str):
self.add_string(constants.KEY_GENERAL_AUTHOR, author) 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): def add_url(self, url: str):
self.add_string(constants.KEY_GENERAL_URL, url) self.add_string(constants.KEY_GENERAL_URL, url)