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