This change modifies the `simulstreaming` backend to support loading fine-tuned Whisper models via the `--model_dir` argument. The `SimulStreamingASR` class has been updated to: - Use the `model_dir` path directly to load the model, which is the correct procedure for fine-tuned `.pt` files. - Automatically disable the `faster-whisper` and `mlx-whisper` fast encoders when `model_dir` is used, as they are not compatible with standard fine-tuned models. The call site in `core.py` already passed the `model_dir` argument, so no changes were needed there. This change makes the `simulstreaming` backend more flexible and allows users to leverage their own custom models. |
||
|---|---|---|
| .. | ||
| whisper | ||
| __init__.py | ||
| backend.py | ||
| beam.py | ||
| config.py | ||
| eow_detection.py | ||
| generation_progress.py | ||
| license_simulstreaming.py | ||
| mlx_encoder.py | ||
| simul_whisper.py | ||
| token_buffer.py | ||