csukuangfj
commited on
Commit
•
9c3d4ca
1
Parent(s):
756d91b
small fixes
Browse files
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=
|
|
|
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 |
|