Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-09-19 01:05:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

use python csv writer for benchmark

Browse Source
This commit is contained in:
Concedo
2026-08-09 09:28:18 +08:00
parent e53ea4a7e4
commit 59d92956f7
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
koboldcpp.py
+5 -3
View File
@@ -12628,11 +12628,13 @@ def kcpp_main_process(launch_args, g_memory=None, gui_launcher=False):
print(f"Output: {result}\n-----")
if save_to_file:
try:
with open(args.benchmark, "a") as file:
import csv
with open(args.benchmark, "a", newline="", encoding="utf-8") as file:
file.seek(0, 2)
writer = csv.writer(file)
if file.tell() == 0: #empty file
file.write("Timestamp,Backend,Layers,Model,MaxCtx,GenAmount,ProcessingTime,ProcessingSpeed,GenerationTime,GenerationSpeed,TotalTime,Output,Flags")
file.write(f"\n{datetimestamp},{libname},{args.gpulayers},{benchmodel},{benchmaxctx},{benchlen},{t_pp:.2f},{s_pp:.2f},{t_gen:.2f},{s_gen:.2f},{(t_pp+t_gen):.2f},{result},\"{benchflagstr}\"")
writer.writerow(["Timestamp","Backend","Layers","Model","MaxCtx","GenAmount","ProcessingTime","ProcessingSpeed","GenerationTime","GenerationSpeed","TotalTime","Output","Flags"])
writer.writerow([datetimestamp,libname,args.gpulayers,benchmodel,benchmaxctx,benchlen,f"{t_pp:.2f}",f"{s_pp:.2f}",f"{t_gen:.2f}",f"{s_gen:.2f}",f"{(t_pp+t_gen):.2f}",result,benchflagstr])
except Exception as e:
print(f"Error writing benchmark to file: {e}")
if global_memory and using_gui_launcher and not save_to_file: