gguf.py : add tensor data layout

This commit is contained in:
klosax 2023-08-15 19:53:44 +02:00 committed by GitHub
parent 2dd5d2c92c
commit b6056c3db8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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)