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:
94
94
print ("gguf: get model metadata" )
95
95
96
96
if "n_layer" in hparams :
97
- block_count = hparams ["n_layer" ]
97
+ block_count = hparams ["n_layer" ]
98
98
elif "num_hidden_layers" in hparams :
99
- block_count = hparams ["num_hidden_layers" ]
99
+ block_count = hparams ["num_hidden_layers" ]
100
100
else :
101
- print ("No block count found" )
101
+ print ("No block count found" )
102
102
103
- sys .exit ()
103
+ sys .exit ()
104
104
105
105
if "n_head" in hparams :
106
- n_head = hparams ["n_head" ]
106
+ n_head = hparams ["n_head" ]
107
107
elif "num_attention_heads" in hparams :
108
- n_head = hparams ["num_attention_heads" ]
108
+ n_head = hparams ["num_attention_heads" ]
109
109
else :
110
- print ("No head count found" )
110
+ print ("No head count found" )
111
111
112
- sys .exit ()
112
+ sys .exit ()
113
113
114
114
n_head_kv = hparams ["n_head_kv" ] if "n_head_kv" in hparams else 1
115
115
You can’t perform that action at this time.
0 commit comments