minor : normalize coding style

pull/294/head
Georgi Gerganov 3 years ago committed by GitHub
parent 9cb835b795
commit 7ef8ab60fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -2357,15 +2357,15 @@ struct whisper_token_data whisper_sample_timestamp(struct whisper_context * ctx,
return res;
}
int whisper_tokenize(struct whisper_context * ctx, const char * text, whisper_token * tokens, std::size_t n_max_tokens) {
int whisper_tokenize(struct whisper_context * ctx, const char * text, whisper_token * tokens, int n_max_tokens) {
const auto res = tokenize(ctx->vocab, text);
if (res.size() > n_max_tokens) {
fprintf(stderr, "%s: too many resulting tokens: %d (max %zu)\n", __func__, (int) res.size(), n_max_tokens);
fprintf(stderr, "%s: too many resulting tokens: %d (max %d)\n", __func__, (int) res.size(), n_max_tokens);
return -1;
}
for (std::size_t i = 0; i < res.size(); i++) {
for (int i = 0; i < (int) res.size(); i++) {
tokens[i] = res[i];
}
@ -2438,7 +2438,7 @@ int whisper_lang_auto_detect(
}
std::vector<std::pair<float, int>> probs_id;
for (const auto &kv : g_lang) {
for (const auto & kv : g_lang) {
const auto token_lang = whisper_token_lang(ctx, kv.second.first);
probs_id.push_back({ ctx->probs[token_lang], kv.second.first });
}
@ -2464,7 +2464,7 @@ int whisper_lang_auto_detect(
}
{
for (std::size_t i = 0; i < probs_id.size(); i++) {
for (int i = 0; i < (int) probs_id.size(); i++) {
if (lang_probs) {
lang_probs[probs_id[i].second] = probs_id[i].first;
}

@ -1,7 +1,6 @@
#ifndef WHISPER_H
#define WHISPER_H
#include <cstddef>
#include <stdint.h>
#include <stdbool.h>
@ -149,7 +148,7 @@ extern "C" {
struct whisper_context * ctx,
const char * text,
whisper_token * tokens,
std::size_t n_max_tokens);
int n_max_tokens);
// Largest language id (i.e. number of available languages - 1)
WHISPER_API int whisper_lang_max_id();

Loading…
Cancel
Save