vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Call progress callback more frequently

Browse source
This commit is contained in:
Jed Fox 2023-03-24 17:26:19 -04:00
parent 1e3fd898a3
commit ae3d0ff68f
No known key found for this signature in database
GPG key ID: 0B61D18EA54B47E1
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama.cpp
View file

@ -776,11 +776,12 @@ static bool llama_model_load(
model.n_loaded++; model.n_loaded++;
// progress // progress
if (progress_callback) {
double current_file_progress = double(size_t(fin.tellg()) - file_offset) / double(file_size - file_offset);
double current_progress = (double(i) + current_file_progress) / double(n_parts);
progress_callback(current_progress, progress_callback_user_data);
}
if (model.n_loaded % 8 == 0) { if (model.n_loaded % 8 == 0) {
if (progress_callback) {
double current_progress = (double(i) + (double(size_t(fin.tellg()) - file_offset) / double(file_size - file_offset))) / double(n_parts);
progress_callback(current_progress, progress_callback_user_data);
}
fprintf(stderr, "."); fprintf(stderr, ".");
fflush(stderr); fflush(stderr);
} }