Skip to content

Commit e5c5fd7

Browse files
authored
Update convert_hf_to_gguf.py
1 parent c3dbfb6 commit e5c5fd7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -460,7 +460,7 @@ def __init__(self, *args, **kwargs):
460460
# move the text_config to the root level
461461
self.hparams = {**self.hparams, **self.hparams["text_config"]}
462462

463-
self.block_count = self.find_hparam(["n_layers", "num_hidden_layers", "n_layer", "num_layers"])
463+
self.block_count = self.find_hparam(["n_layers", "num_hidden_layers", "n_layer", "num_layers", "depth"])
464464
self.tensor_map = gguf.get_tensor_name_map(self.model_arch, self.block_count)
465465

466466
@classmethod

0 commit comments

Comments
 (0)