From d2ce9cfe8d0c75ef4166bac28e93854319219390 Mon Sep 17 00:00:00 2001 From: klosax <131523366+klosax@users.noreply.github.com> Date: Sun, 13 Aug 2023 00:01:20 +0200 Subject: [PATCH] gguf.py : n_layer --> n_block --- gguf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gguf.py b/gguf.py index 0854418a6..75ba2870a 100644 --- a/gguf.py +++ b/gguf.py @@ -253,9 +253,9 @@ class GGUFWriter: self.add_uint32( constants.KEY_LLM_EMBEDDING_LENGTH.format(llm=llm), length) - def add_layer_count(self, llm: str, length: int): + def add_block_count(self, llm: str, length: int): self.add_uint32( - constants.KEY_LLM_LAYER_COUNT.format(llm=llm), length) + constants.KEY_LLM_BLOCK_COUNT.format(llm=llm), length) def add_feed_forward_length(self, llm: str, length: int): self.add_uint32(