Skip to content

Instantly share code, notes, and snippets.

View yy-zhong's full-sized avatar

yy-zhong yy-zhong

  • Tongji University
  • Shanghai, China
  • 09:47 (UTC +08:00)
View GitHub Profile
@yy-zhong
yy-zhong / README.md
Created May 13, 2024 04:34
构建CapsWriter-Offline的新模型

运行python3 download-model.py,在下载目录里将model_quant.onnx复制到根目录,并重命名为model.onnx

运行python3 add-model-metadata.py,此时model.onnx会得到更新。

运行python3 quantize-model.py,获得model.int8.onnx

model.int8.onnxconfig.yamlconfiguration.json复制到原始的models\paraformer-offline-zh中,此时更新完成。

P.S. 原始的tokens.txt无需修改。

# 需要实现备份原先的仓库
import json
import os
import re
import shutil
# 所有出现过的链接,包括没有对应文件的链接
ALL_LINKS = []
# 每个链接对应的文件相对路径