gptc/profiler.py

17 lines
309 B
Python
Raw Normal View History

2023-04-16 14:27:31 -07:00
# SPDX-License-Identifier: GPL-3.0-or-later
import cProfile
import gptc
import json
import sys
max_ngram_length = 10
with open("models/raw.json") as f:
raw_model = json.load(f)
with open("models/benchmark_text.txt") as f:
text = f.read()
cProfile.run("gptc.compile(raw_model, max_ngram_length)")