csukuangfj commited on
Commit
9c3d4ca
1 Parent(s): 756d91b

small fixes

Browse files
Files changed (1) hide show
  1. model.py +3 -1
model.py CHANGED
@@ -18,6 +18,7 @@ import os
18
  from functools import lru_cache
19
  from typing import Union
20
 
 
21
  import torchaudio
22
  from huggingface_hub import hf_hub_download
23
 
@@ -643,7 +644,8 @@ def _get_japanese_pre_trained_model(
643
  tokens=tokens,
644
  use_gpu=False,
645
  feat_config=feat_config,
646
- decoding_method="greedy_search",
 
647
  chunk_size=32,
648
  )
649
 
 
18
  from functools import lru_cache
19
  from typing import Union
20
 
21
+ import torch
22
  import torchaudio
23
  from huggingface_hub import hf_hub_download
24
 
 
644
  tokens=tokens,
645
  use_gpu=False,
646
  feat_config=feat_config,
647
+ decoding_method=decoding_method,
648
+ num_active_paths=num_active_paths,
649
  chunk_size=32,
650
  )
651