File tree Expand file tree Collapse file tree 1 file changed +8
-8
lines changed Expand file tree Collapse file tree 1 file changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -94,22 +94,22 @@ def count_model_parts(dir_model: str) -> int:
9494print ("gguf: get model metadata" )
9595
9696if "n_layer" in hparams :
97- block_count = hparams ["n_layer" ]
97+ block_count = hparams ["n_layer" ]
9898elif "num_hidden_layers" in hparams :
99- block_count = hparams ["num_hidden_layers" ]
99+ block_count = hparams ["num_hidden_layers" ]
100100else :
101- print ("No block count found" )
101+ print ("No block count found" )
102102
103- sys .exit ()
103+ sys .exit ()
104104
105105if "n_head" in hparams :
106- n_head = hparams ["n_head" ]
106+ n_head = hparams ["n_head" ]
107107elif "num_attention_heads" in hparams :
108- n_head = hparams ["num_attention_heads" ]
108+ n_head = hparams ["num_attention_heads" ]
109109else :
110- print ("No head count found" )
110+ print ("No head count found" )
111111
112- sys .exit ()
112+ sys .exit ()
113113
114114n_head_kv = hparams ["n_head_kv" ] if "n_head_kv" in hparams else 1
115115
You can’t perform that action at this time.
0 commit comments