Skip to content

Commit 8360c09

Browse files
fix: use local tokenizer in baseline entigraph
1 parent 67e6c4a commit 8360c09

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

baselines/EntiGraph/entigraph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ class EntiGraph:
7676
max_concurrent: int = 1000
7777

7878
def __post_init__(self):
79-
self.tokenizer = Tokenizer()
79+
self.tokenizer = Tokenizer(model_name=os.getenv("TOKENIZER_MODEL"))
8080

8181
# Initialize specialized clients for different tasks to handle different system prompts and modes
8282
self.client_entities = OpenAIClient(

0 commit comments

Comments
 (0)