feat: use pretty-progress for tensor loading (#516)

This commit is contained in:
stduhpf
2024-12-28 06:14:52 +01:00
committed by GitHub
parent d50473dc49
commit 348a54e34a
2 changed files with 8 additions and 2 deletions

View File

@@ -348,7 +348,7 @@ void pretty_progress(int step, int steps, float time) {
}
}
progress += "|";
printf(time > 1.0f ? "\r%s %i/%i - %.2fs/it" : "\r%s %i/%i - %.2fit/s",
printf(time > 1.0f ? "\r%s %i/%i - %.2fs/it" : "\r%s %i/%i - %.2fit/s\033[K",
progress.c_str(), step, steps,
time > 1.0f || time == 0 ? time : (1.0f / time));
fflush(stdout); // for linux