Skip to content

Commit e4b697c

Browse files
committed
Tidy
1 parent a675b2a commit e4b697c

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

Libraries/MLXLMCommon/PromptCache.swift

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -46,10 +46,10 @@ public actor PromptCache {
4646
print("[getUncachedSuffix] comPrefixLength: \(comPrefixLength)")
4747

4848
if comPrefixLength == self.tokens.size {
49-
let optPrompt = prompt[comPrefixLength..<prompt.size]
49+
let suffix = prompt[comPrefixLength..<prompt.size]
5050
print("Concating...")
51-
self.tokens = concatenated([self.tokens, optPrompt], axis: 0)
52-
return optPrompt
51+
self.tokens = concatenated([self.tokens, suffix], axis: 0)
52+
return suffix
5353
} else if (comPrefixLength < self.tokens.size) {
5454
if isTrimmable() {
5555
print("trimming: \(self.tokens.size - comPrefixLength)")
@@ -59,9 +59,9 @@ public actor PromptCache {
5959
print("Warning: request trimmed amount and actual trimmed amount are different")
6060
}
6161
self.tokens = self.tokens[0..<comPrefixLength]
62-
let optPrompt = prompt[comPrefixLength..<prompt.size]
63-
self.tokens = concatenated([self.tokens, optPrompt], axis: 0)
64-
return optPrompt
62+
let suffix = prompt[comPrefixLength..<prompt.size]
63+
self.tokens = concatenated([self.tokens, suffix], axis: 0)
64+
return suffix
6565
} else {
6666
// Caller must create a new cache
6767
return nil

0 commit comments

Comments
 (0)