Skip to content

Commit fe695c9

Browse files
shoumikhinkeyprocedure
authored andcommitted
Make llama model search case insensitive for benchmark app (pytorch#10133)
1 parent 3849366 commit fe695c9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extension/benchmark/apple/Benchmark/Tests/LLaMA/LLaMATests.mm

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ @implementation LLaMATests
6060
+ (NSDictionary<NSString *, BOOL (^)(NSString *)> *)predicates {
6161
return @{
6262
@"model" : ^BOOL(NSString *filename){
63-
return [filename hasSuffix:@".pte"] && [filename containsString:@"llama"];
63+
return [filename hasSuffix:@".pte"] && [filename.lowercaseString containsString:@"llama"];
6464
},
6565
@"tokenizer" : ^BOOL(NSString *filename) {
6666
return [filename isEqual:@"tokenizer.bin"] || [filename isEqual:@"tokenizer.model"];

0 commit comments

Comments
 (0)