bug: bench command fails with error.UseForwardPrefillDecode #55
Loading…
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
The
benchsubcommand insrc/bench/bench.zigcallsmodel.forward()which returnserror.UseForwardPrefillDecode. It should useforward_prefill()andforward_decode()instead.Reproduction
Result:
Error: UseForwardPrefillDecodeExpected
Benchmark should run and report prefill + decode tok/s.
Fix
Update
bench_prefillto callmodel.forward_prefill()andbench_decodeto callmodel.forward_decode()with proper KV-cache management.