diff --git a/extension/benchmark/apple/Benchmark/Tests/LLaMA/LLaMATests.mm b/extension/benchmark/apple/Benchmark/Tests/LLaMA/LLaMATests.mm index 985f77956b6..fe5e816d6a7 100644 --- a/extension/benchmark/apple/Benchmark/Tests/LLaMA/LLaMATests.mm +++ b/extension/benchmark/apple/Benchmark/Tests/LLaMA/LLaMATests.mm @@ -63,7 +63,7 @@ @implementation LLaMATests return [filename hasSuffix:@".pte"] && [filename.lowercaseString containsString:@"llama"]; }, @"tokenizer" : ^BOOL(NSString *filename) { - return [filename isEqual:@"tokenizer.bin"] || [filename isEqual:@"tokenizer.model"]; + return [filename isEqual:@"tokenizer.bin"] || [filename isEqual:@"tokenizer.model"] || [filename isEqual:@"tokenizer.json"]; }, }; }