alvanlii commited on
Commit
029960a
1 Parent(s): 09d91ba

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -4
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)