cutom alignment heads parameter for custom models
This commit is contained in:
commit
d080d675a8
1 changed files with 1 additions and 1 deletions
|
|
@ -206,7 +206,7 @@ class SimulStreamingASR():
|
||||||
self.model_path = os.path.dirname(os.path.abspath(self.cfg.model_path))
|
self.model_path = os.path.dirname(os.path.abspath(self.cfg.model_path))
|
||||||
|
|
||||||
self.mlx_encoder, self.fw_encoder = None, None
|
self.mlx_encoder, self.fw_encoder = None, None
|
||||||
if not self.disable_fast_encoder and not self.model_dir:
|
if not self.disable_fast_encoder:
|
||||||
if HAS_MLX_WHISPER:
|
if HAS_MLX_WHISPER:
|
||||||
print('Simulstreaming will use MLX whisper for a faster encoder.')
|
print('Simulstreaming will use MLX whisper for a faster encoder.')
|
||||||
mlx_model_name = mlx_model_mapping[self.model_name]
|
mlx_model_name = mlx_model_mapping[self.model_name]
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue