Show subcommand: print total parameters.

This commit is contained in:
antirez
2023-12-28 16:06:52 +01:00
parent e2062eea2c
commit 2a599dc5d0
2 changed files with 6 additions and 2 deletions

View File

@@ -163,6 +163,7 @@ void gguf_tools_show(const char *filename) {
/* Show all the tensors. */
gguf_tensor tensor;
uint64_t params = 0;
while (gguf_get_tensor(ctx,&tensor)) {
printf("%s tensor %.*s @%llu, %llu weights, %llu bytes\n",
gguf_get_tensor_type_name(tensor.type),
@@ -171,7 +172,10 @@ void gguf_tools_show(const char *filename) {
tensor.offset,
tensor.num_weights,
tensor.bsize);
params += tensor.num_weights;
}
printf("gguf-tools.info.parameters: %.02fB\n",
(double)params/1000000000);
return;
}