add model_flops

2024-11-21 20:06:16 +04:00
parent 80f6b72e71
commit 7ee1423006
5 changed files with 67 additions and 42 deletions
@@ -914,7 +914,7 @@ struct llama_init_result llama_init_from_gpt_params(gpt_params & params) {
        dev_info_set = (struct device_info *)malloc(n_world * sizeof(struct device_info));
        dev_info_set[0] = dev_info;
        llama_gather_device_info(lctx, dev_info_set);
-        device_print_props(dev_info_set, n_world);
+        device_print_props(dev_info_set, n_world, model);
    } else {
        llama_send_device_info(lctx, &dev_info);
    }