bench : variable n_past

This commit is contained in:
Georgi Gerganov 2023-09-14 22:41:41 +03:00
parent bfcb2a2ab9
commit a1664574fe
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -105,8 +105,8 @@ int whisper_bench_full(const whisper_params & params) {
}
}
for (int i = 0; i < 128; i++) {
if (int ret = whisper_decode(ctx, tokens, 1, 256, params.n_threads) != 0) {
for (int i = 0; i < 256; i++) {
if (int ret = whisper_decode(ctx, tokens, 1, i, params.n_threads) != 0) {
fprintf(stderr, "error: failed to encode model: %d\n", ret);
return 4;
}