yum-archive/TaSTT-Whisper

High-performance GPGPU inference of OpenAI's Whisper automatic speech recognition (ASR) model

git clone https://git.yummers.dev/yum-archive/TaSTT-Whisper

KonstantinSource codes8c4603c

master
branchesmasterall refs →
844 B38 linesraw
1#pragma once
2#include <vector>
3#include <string>
4
5// command-line parameters
6struct whisper_params
7{
8	uint32_t n_threads;
9	uint32_t n_processors = 1;
10	uint32_t offset_t_ms = 0;
11	uint32_t offset_n = 0;
12	uint32_t duration_ms = 0;
13	uint32_t max_context = UINT_MAX;
14	uint32_t max_len = 0;
15
16	float word_thold = 0.01f;
17
18	bool speed_up = false;
19	bool translate = false;
20	bool diarize = false;
21	bool output_txt = false;
22	bool output_vtt = false;
23	bool output_srt = false;
24	bool output_wts = false;
25	bool print_special = false;
26	bool print_colors = true;
27	bool no_timestamps = false;
28
29	std::string language = "en";
30	std::wstring model = L"models/ggml-base.en.bin";
31	std::vector<std::wstring> fname_inp;
32
33	whisper_params();
34
35	bool parse( int argc, wchar_t* argv[] );
36};
37
38void whisper_print_usage( int argc, wchar_t** argv, const whisper_params& params );