espnet / espnet_onnx

Onnx wrapper for espnet infrernce model
MIT License
149 stars 24 forks source link

ModuleNotFoundError: No module named 'espnet_model_zoo.downloader'; 'espnet_model_zoo' is not a package #93

Open neso613 opened 1 year ago

neso613 commented 1 year ago

Trying to download the pretrained model using following code -

`from espnet2.bin.asr_inference import Speech2Text from espnet_onnx.export import ASRModelExport

m = ASRModelExport()

tag_name = 'asr_train_asr_transformer2_raw_zh_char_batch_bins20000000_ctc_confignore_nan_gradtrue_sp_valid.acc.ave'

download with espnet_model_zoo and export from pretrained model

m.export_from_pretrained(tag_name, quantize=True)

export from trained model

speech2text = Speech2Text(args) m.export(speech2text, '', quantize=True)`

and ends up with the error

Traceback (most recent call last): File "espnet_onnx_export.py", line 7, in <module> from espnet_onnx.export import ASRModelExport File "/home/neso/espnet_onnx/espnet_onnx/export/__init__.py", line 1, in <module> from .asr.export_asr import ASRModelExport File "/home/neso/espnet_onnx/espnet_onnx/export/asr/export_asr.py", line 14, in <module> from espnet_model_zoo.downloader import ModelDownloader File "/home/neso/espnet_onnx/espnet_model_zoo.py", line 8, in <module> speech2text = Speech2Text.from_pretrained( File "/home/neso/anaconda3/envs/espnet_onnx/lib/python3.8/site-packages/espnet2/bin/asr_inference.py", line 556, in from_pretrained from espnet_model_zoo.downloader import ModelDownloader ModuleNotFoundError: No module named 'espnet_model_zoo.downloader'; 'espnet_model_zoo' is not a package

Masao-Someki commented 1 year ago

Hi @neso613, please install espnet_model_zoo. You can install it via pip install espnet_model_zoo