From 111450d5e67ed01e3e7de088fe5bb3115dc7eeff Mon Sep 17 00:00:00 2001
From: mengzhe.cmz <mengzhe.cmz@alibaba-inc.com>
Date: 星期四, 23 三月 2023 19:55:15 +0800
Subject: [PATCH] update infer.py
---
funasr/runtime/onnxruntime/src/Model.cpp | 5 ++---
1 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/Model.cpp b/funasr/runtime/onnxruntime/src/Model.cpp
index ddd4fd0..7ddb635 100644
--- a/funasr/runtime/onnxruntime/src/Model.cpp
+++ b/funasr/runtime/onnxruntime/src/Model.cpp
@@ -1,11 +1,10 @@
#include "precomp.h"
-Model *create_model(const char *path,int nThread)
+Model *create_model(const char *path, int nThread, bool quantize)
{
Model *mm;
-
- mm = new paraformer::ModelImp(path, nThread);
+ mm = new paraformer::ModelImp(path, nThread, quantize);
return mm;
}
--
Gitblit v1.9.1