Преглед изворни кода

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

onnx export
zhifu gao пре 3 година
родитељ
комит
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 numpy as np
 import random
 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:
 class ASRModelExportParaformer:
     def __init__(self, cache_dir: Union[Path, str] = None, onnx: bool = True):
     def __init__(self, cache_dir: Union[Path, str] = None, onnx: bool = True):