bench : fix build + fix go bindings

This commit is contained in:
Georgi Gerganov 2023-11-07 13:20:02 +02:00
parent 185d3fd6d9
commit 8fb0a1cd1c
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
3 changed files with 6 additions and 3 deletions

View File

@ -83,7 +83,6 @@ const (
SampleRate = C.WHISPER_SAMPLE_RATE // Expected sample rate, samples per second
SampleBits = uint16(unsafe.Sizeof(C.float(0))) * 8 // Sample size in bits
NumFFT = C.WHISPER_N_FFT
NumMEL = C.WHISPER_N_MEL
HopLength = C.WHISPER_HOP_LENGTH
ChunkSize = C.WHISPER_CHUNK_SIZE
)

View File

@ -23,7 +23,9 @@ void bench_main(size_t index) {
fprintf(stderr, "%s: running benchmark with %d threads - please wait...\n", __func__, n_threads);
if (int ret = whisper_set_mel(ctx, nullptr, 0, WHISPER_N_MEL)) {
const int n_mel = 80;
if (int ret = whisper_set_mel(ctx, nullptr, 0, n_mel)) {
fprintf(stderr, "error: failed to set mel: %d\n", ret);
return;
}

View File

@ -73,7 +73,9 @@ int whisper_bench_full(const whisper_params & params) {
return 2;
}
if (int ret = whisper_set_mel(ctx, nullptr, 0, WHISPER_N_MEL)) {
const int n_mel = 80;
if (int ret = whisper_set_mel(ctx, nullptr, 0, n_mel)) {
fprintf(stderr, "error: failed to set mel: %d\n", ret);
return 3;
}