Merge branch 'main' into dev_onnx

Xian Shi

2023-10-17 a05e753d11d9c36983ec4e58c421dbcf86d1dcd4

 funasr/utils/runtime_sdk_download_tool.py

@@ -6,6 +6,7 @@
parser = argparse.ArgumentParser()
parser.add_argument('--model-name', type=str, required=True)
parser.add_argument('--export-dir', type=str, required=True)
parser.add_argument('--export', type=str2bool, default=True, help='whether to export model')
parser.add_argument('--type', type=str, default='onnx', help='["onnx", "torch"]')
parser.add_argument('--device', type=str, default='cpu', help='["cpu", "cuda"]')
parser.add_argument('--quantize', type=str2bool, default=False, help='export quantized model')
@@ -23,7 +24,7 @@
   except:
      raise "model_dir must be model_name in modelscope or local path downloaded from modelscope, but is {}".format \
         (model_dir)

if args.export:
model_file = os.path.join(model_dir, 'model.onnx')
if args.quantize:
   model_file = os.path.join(model_dir, 'model_quant.onnx')