File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change @@ -147,6 +147,8 @@ class Task(enum.StrEnum):
147
147
148
148
149
149
class WhisperConfig (BaseModel ):
150
+ """See https://github.com/SYSTRAN/faster-whisper/blob/master/faster_whisper/transcribe.py#L599."""
151
+
150
152
model : str = Field (default = "Systran/faster-whisper-medium.en" )
151
153
"""
152
154
Huggingface model to use for transcription. Note, the model must support being ran using CTranslate2.
@@ -155,6 +157,9 @@ class WhisperConfig(BaseModel):
155
157
"""
156
158
inference_device : Device = Field (default = Device .AUTO )
157
159
compute_type : Quantization = Field (default = Quantization .DEFAULT )
160
+ device_index : int | list [int ] = 0
161
+ cpu_threads : int = 16
162
+ num_workers : int = 1
158
163
159
164
160
165
class Config (BaseSettings ):
You can’t perform that action at this time.
0 commit comments