Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
neso613 opened this issue Jun 8, 2023 · 1 comment

Comments

@neso613
Copy link

neso613 commented Jun 8, 2023

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
Copy link
Collaborator

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants