Skip to content

Commit e00177a

Browse files
authored
Update LLM generation config initialization
1 parent 59269a6 commit e00177a

File tree

1 file changed

+1
-1
lines changed
  • extension/benchmark/apple/Benchmark/Tests/LLM

1 file changed

+1
-1
lines changed

extension/benchmark/apple/Benchmark/Tests/LLM/LLMTests.mm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ @implementation LLMTests
102102
block:^{
103103
tokensPerSecondMetric.tokenCount = 0;
104104
BOOL status = [runner generateWithPrompt:@"Once upon a time"
105-
config:[[ExecuTorchLLMGenerationConfig alloc] initWithBlock:^(ExecuTorchLLMGenerationConfig *config) {
105+
config:[[ExecuTorchLLMConfig alloc] initWithBlock:^(ExecuTorchLLMConfig *config) {
106106
config.sequenceLength = 50;
107107
}]
108108
tokenCallback:^(NSString *token) {

0 commit comments

Comments
 (0)