Skip to content

Commit

Permalink
Merge pull request #19 from lean-dojo/dev
Browse files Browse the repository at this point in the history
minor change
Kaiyu Yang authored Nov 12, 2023
2 parents 9d6f4c1 + 852465b commit 8f926b0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion LeanInfer/Basic.lean
Original file line number Diff line number Diff line change
@@ -65,7 +65,7 @@ def generate (input : String) (targetPrefix : String) : m (Array (String × Floa
let temperature := config.decoding.temperature
let beamSize := config.decoding.beamSize
let rawOutputs := FFI.onnxGenerate input numReturnSequences maxLength temperature beamSize
rawOutputs.filter (entry : String × Float) => entry.fst ≠ "aesop")
rawOutputs.filter fun (entry : String × Float) => entry.fst ≠ "aesop"
| .native (.ct2 _) =>
let inputTokens := tokenizeByt5 input true |>.toArray
let targetPrefixTokens := tokenizeByt5 targetPrefix false |>.toArray

0 comments on commit 8f926b0

Please sign in to comment.