From cfc4d402093060fe087424b0a6be4e2b2546eae8 Mon Sep 17 00:00:00 2001
From: wanchen.swc <wanchen.swc@alibaba-inc.com>
Date: 星期四, 30 三月 2023 18:15:15 +0800
Subject: [PATCH] [Export] support gpu inference

---
 funasr/export/test_torchscripts.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/export/test_torchscripts.py b/funasr/export/test_torchscripts.py
index 11be763..9afec74 100644
--- a/funasr/export/test_torchscripts.py
+++ b/funasr/export/test_torchscripts.py
@@ -2,7 +2,7 @@
 import numpy as np
 
 if __name__ == '__main__':
-	onnx_path = "/mnt/workspace/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/model.torchscripts"
+	onnx_path = "/nfs/zhifu.gzf/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/model.torchscripts"
 	loaded = torch.jit.load(onnx_path)
 	
 	x = torch.rand([2, 21, 560])

--
Gitblit v1.9.1