Просмотр исходного кода

llama : add log about loading model tensors (#11699)

Georgi Gerganov 11 месяцев назад
Родитель
Сommit
9dd7a0390f
1 измененных файлов с 2 добавлено и 0 удалено
  1. 2 0
      src/llama-model.cpp

+ 2 - 0
src/llama-model.cpp

@@ -1275,6 +1275,8 @@ bool llama_model::load_tensors(llama_model_loader & ml) {
 
     const bool use_mmap_buffer = true;
 
+    LLAMA_LOG_INFO("%s: loading model tensors, this can take a while... (mmap = %s)\n", __func__, use_mmap_buffer ? "true" : "false");
+
     // build a list of buffer types for the CPU and GPU devices
     pimpl->cpu_buft_list = make_cpu_buft_list(devices);
     for (auto * dev : devices) {