feat: add large-v3 model

This commit is contained in:
华晨
2023-12-30 23:19:30 +08:00
parent 5de4de4b84
commit 71d207a607
4 changed files with 6 additions and 4 deletions
+1
View File
@@ -140,6 +140,7 @@
<option value="small" selected>Small</option>
<option value="medium">Medium</option>
<option value="large-v2">Large-v2</option>
<option value="large-v3">Large-v3</option>
</select>
</div>
</body>
+1
View File
@@ -142,6 +142,7 @@
<option value="small" selected>Small</option>
<option value="medium">Medium</option>
<option value="large-v2">Large-v2</option>
<option value="large-v3">Large-v3</option>
</select>
</div>
</body>
+2 -2
View File
@@ -211,7 +211,7 @@ class ServeClient:
self.data = b""
self.frames = b""
self.model_sizes = [
"tiny", "base", "small", "medium", "large-v2"
"tiny", "base", "small", "medium", "large-v2", "large-v3"
]
self.multilingual = multilingual
self.model_size = self.get_model_size(model_size)
@@ -277,7 +277,7 @@ class ServeClient:
)
return None
if model_size == "large-v2":
if model_size in ["large-v2", "large-v3"]:
self.multilingual = True
return model_size
+2 -2
View File
@@ -94,7 +94,7 @@ class WhisperModel:
Args:
model_size_or_path: Size of the model to use (tiny, tiny.en, base, base.en,
small, small.en, medium, medium.en, large-v1, large-v2, or large), a path to a converted
small, small.en, medium, medium.en, large-v1, large-v2, large-v3, or large), a path to a converted
model directory, or a CTranslate2-converted Whisper model ID from the Hugging Face Hub.
When a size or a model ID is configured, the converted model is downloaded
from the Hugging Face Hub.
@@ -914,7 +914,7 @@ class WhisperModel:
words, word_tokens, start_times, end_times, word_probabilities
)
]
def destroy(self):
del self.model