Skip to content

Commit 8f926b0

Browse files
author
Kaiyu Yang
authored
Merge pull request #19 from lean-dojo/dev
minor change
2 parents 9d6f4c1 + 852465b commit 8f926b0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

LeanInfer/Basic.lean

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def generate (input : String) (targetPrefix : String) : m (Array (String × Floa
6565
let temperature := config.decoding.temperature
6666
let beamSize := config.decoding.beamSize
6767
let rawOutputs := FFI.onnxGenerate input numReturnSequences maxLength temperature beamSize
68-
rawOutputs.filter (entry : String × Float) => entry.fst ≠ "aesop")
68+
rawOutputs.filter fun (entry : String × Float) => entry.fst ≠ "aesop"
6969
| .native (.ct2 _) =>
7070
let inputTokens := tokenizeByt5 input true |>.toArray
7171
let targetPrefixTokens := tokenizeByt5 targetPrefix false |>.toArray

0 commit comments

Comments
 (0)