summaryrefslogtreecommitdiffstats
path: root/Whisper/API/SpecialTokens.h
blob: 67fd0209f2937448b971c60ffd50879f811e4921 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#pragma once

namespace Whisper
{
	struct SpecialTokens
	{
		// The end of a transcription, token_eot
		int TranscriptionEnd;
		// Start of a transcription, token_sot
		int TranscriptionStart;
		// Represents the previous word in the transcription. It is used to help the model predict the current word based on the context of the words that came before it.
		int PreviousWord;   // token_prev
		// Start of a sentence
		int SentenceStart;   // token_solm
		//Represents the word "not" in the transcription
		int Not;    // token_not
		//New transcription
		int TranscriptionBegin;    // token_beg

		// token_translate
		int TaskTranslate;
		// token_transcribe
		int TaskTranscribe;
	};
}