turboderp-org/exllamav3
1
0
Fork 0
mirror of https://github.com/turboderp-org/exllamav3.git synced 2026-03-15 00:07:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add optional arg to compare_q to allow it to save plots rather than show them

Browse Source
This commit is contained in:
MikeRoz47
2026-02-15 16:41:18 +00:00
parent 89b841dd8a
commit 52c2f5794d
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
eval/compare_q.py
View File

@@ -317,7 +317,10 @@ def plot(results, args):
plt.grid(color = 'dimgray', linestyle = '--', linewidth = 0.5)
else:
plt.grid(True)
plt.show()
if args.plot_file:
plt.savefig(args.plot_file)
else:
plt.show()
def dict_hash(x: dict) -> str:
@@ -403,6 +406,7 @@ if __name__ == "__main__":
parser.add_argument("-mask", "--mask", type = str, help = "Semicolon-separated list of strings to match against model labels for inclusion")
parser.add_argument("-lf", "--logits_file", type = str, help = "Reference logits file for KLD", required = False)
parser.add_argument("-dark", "--dark", action = "store_true", help = "Dark mode")
parser.add_argument("-pf", "--plot_file", type = str, help = "Write the plot to a file")
_args = parser.parse_args()
main(_args)