Skip to content

Commit c806242

Browse files
authored
Update prompt.py
1 parent b256957 commit c806242

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Needle_test/prompt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def __init__(self,
102102
elif self.tokenizer_type == "Anthropic":
103103
self.enc = Anthropic().get_tokenizer()
104104
elif self.tokenizer_type == "Huggingface":
105-
self.enc = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True)
105+
self.enc = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False)
106106
else:
107107
raise ValueError("tokenizer_type is not supported. Must be either 'tiktoken', 'Anthropic', or 'Huggingface'")
108108

@@ -320,4 +320,4 @@ def start_test(self):
320320
save_dir=config['save_dir'],
321321
)
322322

323-
ht.start_test()
323+
ht.start_test()

0 commit comments

Comments
 (0)