speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

onnx export

Browse Source
This commit is contained in:
游雁 2023-02-27 14:42:13 +08:00
parent f98c4bf6d2
commit 6da6a7847f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/export/export_model.py
View File

@ -11,8 +11,8 @@ from funasr.export.models import get_model
import numpy as np
import random
torch_version = float(".".join(torch.__version__.split(".")[:2]))
assert torch_version > 1.9
# torch_version = float(".".join(torch.__version__.split(".")[:2]))
# assert torch_version > 1.9
class ASRModelExportParaformer:
def __init__(self, cache_dir: Union[Path, str] = None, onnx: bool = True):