Skip to content

Commit e9765bf

Browse files
fix(baselines): fix tokenizer
1 parent 3217b5e commit e9765bf

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

baselines/BDS/bds.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from dotenv import load_dotenv
1010
from tqdm.asyncio import tqdm as tqdm_async
1111

12-
from graphgen.models import NetworkXStorage, OpenAIModel
12+
from graphgen.models import NetworkXStorage, OpenAIModel, Tokenizer
1313
from graphgen.utils import create_event_loop
1414

1515
QA_GENERATION_PROMPT = """
@@ -102,10 +102,14 @@ async def job(item):
102102

103103
load_dotenv()
104104

105+
tokenizer_instance: Tokenizer = Tokenizer(
106+
model_name=os.getenv("TOKENIZER_MODEL", "cl100k_base")
107+
)
105108
llm_client = OpenAIModel(
106109
model_name=os.getenv("SYNTHESIZER_MODEL"),
107110
api_key=os.getenv("SYNTHESIZER_API_KEY"),
108111
base_url=os.getenv("SYNTHESIZER_BASE_URL"),
112+
tokenizer_instance=tokenizer_instance,
109113
)
110114
bds = BDS(llm_client=llm_client)
111115

0 commit comments

Comments
 (0)