Skip to content

Commit 3d59fcb

Browse files
authored
Accept tokenizer.json in iOS benchmark app (#10639)
1 parent 118afa0 commit 3d59fcb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ @implementation LLaMATests
6363
return [filename hasSuffix:@".pte"] && [filename.lowercaseString containsString:@"llama"];
6464
},
6565
@"tokenizer" : ^BOOL(NSString *filename) {
66-
return [filename isEqual:@"tokenizer.bin"] || [filename isEqual:@"tokenizer.model"];
66+
return [filename isEqual:@"tokenizer.bin"] || [filename isEqual:@"tokenizer.model"] || [filename isEqual:@"tokenizer.json"];
6767
},
6868
};
6969
}

0 commit comments

Comments
 (0)