deepgeek/lite-avatar
2
0
Fork 0
mirror of https://github.com/HumanAIGC/lite-avatar.git synced 2026-02-05 18:09:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

add files

Browse Source
This commit is contained in:
烨玮
2025-02-20 12:17:03 +08:00
parent a21dd4555c
commit edd008441b
667 changed files with 473123 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
funasr_local/export/test/test_torchscripts.py Normal file
View File

@@ -0,0 +1,17 @@
import torch
import numpy as np
if __name__ == '__main__':
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])
x_len = torch.IntTensor([6, 21])
res = loaded(x, x_len)
print(res[0].size(), res[1])
x = torch.rand([5, 50, 560])
x_len = torch.IntTensor([6, 21, 10, 30, 50])
res = loaded(x, x_len)
print(res[0].size(), res[1])