added modelName enums
This commit is contained in:
1 parent
7126a66331
commit
d09008827b
1 file changed
+10
@@ -3,6 +3,16 @@ class ModelType(Enum):
|
||||
EMBEDDING = "embedding"
|
||||
TRANSCRIPTION = "transcription"
|
||||
RESPONSE = "response"
|
||||
|
||||
class EmbeddingModelName(Enum):
|
||||
NB_BERT = "nb_bert"
|
||||
|
||||
class TranscriptionModelName(Enum):
|
||||
WHISPER_SMALL = "whisper_small"
|
||||
|
||||
class ResponseModelName(Enum):
|
||||
GENERIC = "generic"
|
||||
|
||||
class PreprocessorType(Enum):
|
||||
NORWEGIAN = "norwegian"
|
||||
DEFAULT = "norwegian"
|
||||
|
||||
Reference in new issue
Block a user