Add support for large-v3 and bump faster-whisper

This commit is contained in:
pluja
2023-11-25 19:58:35 +01:00
parent 1eaf1c42ea
commit 82b531fa45
5 changed files with 6 additions and 3 deletions

1
.gitignore vendored
View File

@@ -10,3 +10,4 @@ whisper_models/
.env .env
.DS_Store .DS_Store
*.beta.yml *.beta.yml
venv

View File

@@ -50,7 +50,8 @@
'small.en', 'small.en',
'medium', 'medium',
'medium.en', 'medium.en',
'large-v2' 'large-v2',
'large-v3'
]; ];
// Sort the languages // Sort the languages
languages.sort((a, b) => { languages.sort((a, b) => {

View File

@@ -3,7 +3,7 @@ from typing import Any, Mapping, TypedDict
import numpy as np import numpy as np
from faster_whisper.audio import decode_audio # type: ignore from faster_whisper.audio import decode_audio # type: ignore
SUPPORTED_MODELS = ["tiny", "tiny.en", "small", "small.en", "base", "base.en", "medium", "medium.en", "large-v2"] SUPPORTED_MODELS = ["tiny", "tiny.en", "small", "small.en", "base", "base.en", "medium", "medium.en", "large-v2", "large-v3"]
WordData = TypedDict( WordData = TypedDict(
"WordData", {"word": str, "start": float | str, "end": float | str, "score": float} "WordData", {"word": str, "start": float | str, "end": float | str, "score": float}

View File

@@ -15,6 +15,7 @@ class ModelSize(str, Enum):
medium_en = "medium.en" medium_en = "medium.en"
medium = "medium" medium = "medium"
large_v2 = "large-v2" large_v2 = "large-v2"
large_v3 = "large-v3"
class Languages(str, Enum): class Languages(str, Enum):
auto = "auto" auto = "auto"

View File

@@ -1,5 +1,5 @@
fastapi==0.100.1 fastapi==0.100.1
faster_whisper==0.9.0 faster-whisper @ https://github.com/guillaumekln/faster-whisper/archive/e1a218fab1ab02d637b79565995bf1a9c4c83a09.tar.gz
pydantic==2.1.1 pydantic==2.1.1
python-dotenv==1.0.0 python-dotenv==1.0.0
uvicorn==0.23.2 uvicorn==0.23.2