Просмотр исходного кода

Merge pull request #157 from alibaba-damo-academy/dev_gzf

onnx export
zhifu gao 3 лет назад
Родитель
Сommit
917b7e03e5
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      funasr/export/export_model.py

+ 2 - 2
funasr/export/export_model.py

@@ -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):