Skip to content

Commit

Permalink
added all the available models
Browse files Browse the repository at this point in the history
  • Loading branch information
joseporiolayats authored Feb 9, 2024
1 parent b1793c9 commit 7064550
Showing 1 changed file with 24 additions and 13 deletions.
37 changes: 24 additions & 13 deletions pywhispercpp/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,19 +25,30 @@

MODELS_DIR = Path(user_data_dir(PACKAGE_NAME)) / 'models'

AVAILABLE_MODELS = ["tiny.en",
"tiny",
"base.en",
"base",
"small.en",
"small",
"medium.en",
"medium",
"large-v1",
"large",
"large-v3",
]

AVAILABLE_MODELS = [
"base",
"base-q5_1",
"base.en",
"base.en-q5_1",
"large-v1",
"large-v2",
"large-v2-q5_0",
"large-v3",
"large-v3-q5_0",
"medium",
"medium-q5_0",
"medium.en",
"medium.en-q5_0",
"small",
"small-q5_1",
"small.en",
"small.en-q5_1",
"tiny",
"tiny-q5_1",
"tiny.en",
"tiny.en-q5_1",
"tiny.en-q8_0",
]
PARAMS_SCHEMA = { # as exactly presented in whisper.cpp
'n_threads': {
'type': int,
Expand Down

0 comments on commit 7064550

Please sign in to comment.