Update README.md
Browse files
README.md
CHANGED
@@ -76,10 +76,6 @@ MODEL_NAME = "alvanlii/whisper-small-cantonese"
|
|
76 |
processor = WhisperProcessor.from_pretrained(MODEL_NAME)
|
77 |
model = WhisperForConditionalGeneration.from_pretrained(MODEL_NAME)
|
78 |
|
79 |
-
model.config.forced_decoder_ids = None
|
80 |
-
model.config.suppress_tokens = None
|
81 |
-
model.config.use_cache = False
|
82 |
-
|
83 |
processed_in = processor(y, sampling_rate=sr, return_tensors="pt")
|
84 |
gout = model.generate(
|
85 |
input_features=processed_in.input_features,
|
@@ -151,6 +147,9 @@ Using speculative decoding with `alvanlii/whisper-small-cantonese`, it runs at 0
|
|
151 |
## Whisper.cpp
|
152 |
Uploaded a GGML bin file for Whisper cpp as of June 2024. You can download the bin file [here](https://huggingface.co/alvanlii/whisper-small-cantonese/blob/main/ggml-model.bin) and try it out [here](https://whisper.ggerganov.com/).
|
153 |
|
|
|
|
|
|
|
154 |
## Training Hyperparameters
|
155 |
- learning_rate: 5e-5
|
156 |
- train_batch_size: 25 (on 1 3090 GPU)
|
|
|
76 |
processor = WhisperProcessor.from_pretrained(MODEL_NAME)
|
77 |
model = WhisperForConditionalGeneration.from_pretrained(MODEL_NAME)
|
78 |
|
|
|
|
|
|
|
|
|
79 |
processed_in = processor(y, sampling_rate=sr, return_tensors="pt")
|
80 |
gout = model.generate(
|
81 |
input_features=processed_in.input_features,
|
|
|
147 |
## Whisper.cpp
|
148 |
Uploaded a GGML bin file for Whisper cpp as of June 2024. You can download the bin file [here](https://huggingface.co/alvanlii/whisper-small-cantonese/blob/main/ggml-model.bin) and try it out [here](https://whisper.ggerganov.com/).
|
149 |
|
150 |
+
## Whisper CT2
|
151 |
+
For use in WhisperX or FasterWhisper, a CT2 file is needed. The converted model is under [here](https://huggingface.co/alvanlii/whisper-small-cantonese/tree/main/cts)
|
152 |
+
|
153 |
## Training Hyperparameters
|
154 |
- learning_rate: 5e-5
|
155 |
- train_batch_size: 25 (on 1 3090 GPU)
|