We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3217b5e commit e9765bfCopy full SHA for e9765bf
1 file changed
baselines/BDS/bds.py
@@ -9,7 +9,7 @@
9
from dotenv import load_dotenv
10
from tqdm.asyncio import tqdm as tqdm_async
11
12
-from graphgen.models import NetworkXStorage, OpenAIModel
+from graphgen.models import NetworkXStorage, OpenAIModel, Tokenizer
13
from graphgen.utils import create_event_loop
14
15
QA_GENERATION_PROMPT = """
@@ -102,10 +102,14 @@ async def job(item):
102
103
load_dotenv()
104
105
+ tokenizer_instance: Tokenizer = Tokenizer(
106
+ model_name=os.getenv("TOKENIZER_MODEL", "cl100k_base")
107
+ )
108
llm_client = OpenAIModel(
109
model_name=os.getenv("SYNTHESIZER_MODEL"),
110
api_key=os.getenv("SYNTHESIZER_API_KEY"),
111
base_url=os.getenv("SYNTHESIZER_BASE_URL"),
112
+ tokenizer_instance=tokenizer_instance,
113
)
114
bds = BDS(llm_client=llm_client)
115
0 commit comments