subaligner.llm module

class subaligner.llm.FacebookMbartFlavour(value)

Bases: enum.Enum

An enumeration.

LARGE = 'large'

class subaligner.llm.HelsinkiNLPFlavour(value)

Bases: enum.Enum

An enumeration.

OPUS_MT = 'Helsinki-NLP/opus-mt-{}-{}'

OPUS_MT_TC_BIG = 'Helsinki-NLP/opus-mt-tc-big-{}-{}'

OPUS_TATOEBA = 'Helsinki-NLP/opus-tatoeba-{}-{}'

class subaligner.llm.TranscriptionRecipe(value)

Bases: enum.Enum

An enumeration.

WHISPER = 'whisper'

class subaligner.llm.TranslationRecipe(value)

Bases: enum.Enum

An enumeration.

FACEBOOK_MBART = 'facebook-mbart'

HELSINKI_NLP = 'helsinki-nlp'

WHISPER = 'whisper'

class subaligner.llm.WhisperFlavour(value)

Bases: enum.Enum

An enumeration.

BASE = 'base'

BASE_EN = 'base.en'

LARGE = 'large'

LARGE_V1 = 'large-v1'

LARGE_V2 = 'large-v2'

MEDIUM = 'medium'

MEDIUM_EN = 'medium.en'

SMALL = 'small'

TINY = 'tiny'

TINY_EN = 'tiny.en'