command : better indentation

pull/285/head
Georgi Gerganov 2 years ago
parent 5eeeb3412d
commit 4312995974

@ -618,7 +618,7 @@ int main(int argc, char ** argv) {
for (int i = 0; i < (int) allowed_commands.size(); ++i) { for (int i = 0; i < (int) allowed_commands.size(); ++i) {
fprintf(stderr, " - \033[1m%-*s\033[0m = [", max_len, allowed_commands[i].c_str()); fprintf(stderr, " - \033[1m%-*s\033[0m = [", max_len, allowed_commands[i].c_str());
for (const auto & token : allowed_tokens[i]) { for (const auto & token : allowed_tokens[i]) {
fprintf(stderr, " %d", token); fprintf(stderr, " %5d", token);
} }
fprintf(stderr, " ]\n"); fprintf(stderr, " ]\n");
} }
@ -835,7 +835,7 @@ int main(int argc, char ** argv) {
for (const auto & cmd : probs_id) { for (const auto & cmd : probs_id) {
fprintf(stdout, "%s: %s%-*s%s = %f | ", __func__, "\033[1m", max_len, allowed_commands[cmd.second].c_str(), "\033[0m", cmd.first); fprintf(stdout, "%s: %s%-*s%s = %f | ", __func__, "\033[1m", max_len, allowed_commands[cmd.second].c_str(), "\033[0m", cmd.first);
for (int i = 0; i < (int) allowed_tokens[cmd.second].size(); ++i) { for (int i = 0; i < (int) allowed_tokens[cmd.second].size(); ++i) {
fprintf(stdout, "%f ", probs[allowed_tokens[cmd.second][i]]); fprintf(stdout, "'%4s' %f ", whisper_token_to_str(ctx, allowed_tokens[cmd.second][i]), probs[allowed_tokens[cmd.second][i]]);
} }
fprintf(stdout, "\n"); fprintf(stdout, "\n");
} }

Loading…
Cancel
Save