zhifu gao
2023-02-27 917b7e03e5c1c9719323926f9595666abec17128
Merge pull request #157 from alibaba-damo-academy/dev_gzf

onnx export
1个文件已修改
4 ■■■■ 已修改文件
funasr/export/export_model.py 4 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/export/export_model.py
@@ -11,8 +11,8 @@
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):