We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b256957 commit c806242Copy full SHA for c806242
Needle_test/prompt.py
@@ -102,7 +102,7 @@ def __init__(self,
102
elif self.tokenizer_type == "Anthropic":
103
self.enc = Anthropic().get_tokenizer()
104
elif self.tokenizer_type == "Huggingface":
105
- self.enc = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True)
+ self.enc = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False)
106
else:
107
raise ValueError("tokenizer_type is not supported. Must be either 'tiktoken', 'Anthropic', or 'Huggingface'")
108
@@ -320,4 +320,4 @@ def start_test(self):
320
save_dir=config['save_dir'],
321
)
322
323
- ht.start_test()
+ ht.start_test()
0 commit comments