Skip to content

Commit cf59429

Browse files
committed
Use faster-whisper model names instead of hardcoded HF names
1 parent 8e47917 commit cf59429

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

ctranslate2/calc_rtf.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,13 @@
77
device_index = 0
88

99
models = [
10-
"guillaumekln/faster-whisper-tiny.en",
11-
"guillaumekln/faster-whisper-small.en",
12-
"guillaumekln/faster-whisper-base.en",
13-
"guillaumekln/faster-whisper-medium.en",
14-
"guillaumekln/faster-whisper-large-v1",
15-
"guillaumekln/faster-whisper-large-v2",
10+
"tiny.en",
11+
"small.en",
12+
"base.en",
13+
"medium.en",
14+
"large-v1",
15+
"large-v2",
16+
"large-v3"
1617
]
1718

1819
n_batches = 3

ctranslate2/run_whisper.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
export PYTHONPATH="..":$PYTHONPATH
44

5-
MODEL_IDs=("guillaumekln/faster-whisper-tiny.en" "guillaumekln/faster-whisper-small.en" "guillaumekln/faster-whisper-base.en" "guillaumekln/faster-whisper-medium.en" "guillaumekln/faster-whisper-large-v1" "guillaumekln/faster-whisper-large-v2")
5+
MODEL_IDs=("tiny.en" "small.en" "base.en" "medium.en" "large-v1" "large-v2" "large-v3")
66
BATCH_SIZE=1
77
DEVICE_INDEX=0
88

0 commit comments

Comments
 (0)