Skip to content

Commit 284ceb3

Browse files
DamonFoolCISC
andauthored
Update gguf-py/gguf/gguf_writer.py
Co-authored-by: Sigbjørn Skjæret <[email protected]>
1 parent 2161c30 commit 284ceb3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

gguf-py/gguf/gguf_writer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -676,8 +676,8 @@ def add_parallel_residual(self, use: bool) -> None:
676676
def add_decoder_start_token_id(self, id: int) -> None:
677677
self.add_uint32(Keys.LLM.DECODER_START_TOKEN_ID.format(arch=self.arch), id)
678678

679-
def add_num_decoder_layers(self, value: int) -> None:
680-
self.add_uint32(Keys.LLM.NUM_DECODER_LAYERS.format(arch=self.arch), value)
679+
def add_decoder_block_count(self, value: int) -> None:
680+
self.add_uint32(Keys.LLM.DECODER_BLOCK_COUNT.format(arch=self.arch), value)
681681

682682
def add_embedding_length_per_layer_input(self, value: int) -> None:
683683
self.add_uint32(Keys.LLM.EMBD_LENGTH_PER_LAYER_INP.format(arch=self.arch), value)

0 commit comments

Comments
 (0)