Fix profiler and benchmark

This commit is contained in:
Samuel Sloniker 2023-04-17 21:28:24 -07:00
parent 7b7ef39d0b
commit 4546c4cffa
Signed by: kj7rrv
GPG Key ID: 1BB4029E66285A62
2 changed files with 2 additions and 2 deletions

View File

@ -25,7 +25,7 @@ print(
round( round(
1000000 1000000
* timeit.timeit( * timeit.timeit(
"gptc.compile(raw_model, max_ngram_length)", "gptc.Model.compile(raw_model, max_ngram_length)",
number=compile_iterations, number=compile_iterations,
globals=globals(), globals=globals(),
) )

View File

@ -13,4 +13,4 @@ with open("models/raw.json") as f:
with open("models/benchmark_text.txt") as f: with open("models/benchmark_text.txt") as f:
text = f.read() text = f.read()
cProfile.run("gptc.compile(raw_model, max_ngram_length)") cProfile.run("gptc.Model.compile(raw_model, max_ngram_length)")