feat: add large-v3 model
This commit is contained in:
@@ -140,6 +140,7 @@
|
|||||||
<option value="small" selected>Small</option>
|
<option value="small" selected>Small</option>
|
||||||
<option value="medium">Medium</option>
|
<option value="medium">Medium</option>
|
||||||
<option value="large-v2">Large-v2</option>
|
<option value="large-v2">Large-v2</option>
|
||||||
|
<option value="large-v3">Large-v3</option>
|
||||||
</select>
|
</select>
|
||||||
</div>
|
</div>
|
||||||
</body>
|
</body>
|
||||||
|
|||||||
@@ -142,6 +142,7 @@
|
|||||||
<option value="small" selected>Small</option>
|
<option value="small" selected>Small</option>
|
||||||
<option value="medium">Medium</option>
|
<option value="medium">Medium</option>
|
||||||
<option value="large-v2">Large-v2</option>
|
<option value="large-v2">Large-v2</option>
|
||||||
|
<option value="large-v3">Large-v3</option>
|
||||||
</select>
|
</select>
|
||||||
</div>
|
</div>
|
||||||
</body>
|
</body>
|
||||||
|
|||||||
@@ -211,7 +211,7 @@ class ServeClient:
|
|||||||
self.data = b""
|
self.data = b""
|
||||||
self.frames = b""
|
self.frames = b""
|
||||||
self.model_sizes = [
|
self.model_sizes = [
|
||||||
"tiny", "base", "small", "medium", "large-v2"
|
"tiny", "base", "small", "medium", "large-v2", "large-v3"
|
||||||
]
|
]
|
||||||
self.multilingual = multilingual
|
self.multilingual = multilingual
|
||||||
self.model_size = self.get_model_size(model_size)
|
self.model_size = self.get_model_size(model_size)
|
||||||
@@ -277,7 +277,7 @@ class ServeClient:
|
|||||||
)
|
)
|
||||||
return None
|
return None
|
||||||
|
|
||||||
if model_size == "large-v2":
|
if model_size in ["large-v2", "large-v3"]:
|
||||||
self.multilingual = True
|
self.multilingual = True
|
||||||
return model_size
|
return model_size
|
||||||
|
|
||||||
|
|||||||
@@ -94,7 +94,7 @@ class WhisperModel:
|
|||||||
|
|
||||||
Args:
|
Args:
|
||||||
model_size_or_path: Size of the model to use (tiny, tiny.en, base, base.en,
|
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.
|
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
|
When a size or a model ID is configured, the converted model is downloaded
|
||||||
from the Hugging Face Hub.
|
from the Hugging Face Hub.
|
||||||
|
|||||||
Reference in New Issue
Block a user