Commit
·
4850190
1
Parent(s):
108659c
small fixes
Browse files
model.py
CHANGED
|
@@ -18,6 +18,7 @@ import os
|
|
| 18 |
from functools import lru_cache
|
| 19 |
from typing import Union
|
| 20 |
|
|
|
|
| 21 |
import torch
|
| 22 |
import torchaudio
|
| 23 |
from huggingface_hub import hf_hub_download
|
|
@@ -2083,16 +2084,28 @@ def _get_funasr_nano(
|
|
| 2083 |
local_dir = snapshot_download(repo_id)
|
| 2084 |
else:
|
| 2085 |
local_dir = "sherpa-onnx-funasr-nano-int8-2025-12-30"
|
| 2086 |
-
os.system(
|
| 2087 |
-
f"git lfs install; git clone https://huggingface.co/csukuangfj/sherpa-onnx-funasr-nano-int8-2025-12-30 {local_dir}"
|
| 2088 |
-
)
|
| 2089 |
|
| 2090 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2091 |
|
| 2092 |
-
|
| 2093 |
-
llm = f"{local_dir}/llm.int8.onnx"
|
| 2094 |
-
embedding = f"{local_dir}/embedding.int8.onnx"
|
| 2095 |
-
tokenizer = f"{local_dir}/Qwen3-0.6B"
|
| 2096 |
|
| 2097 |
print("local_dir", local_dir)
|
| 2098 |
print("encoder_adaptor", encoder_adaptor)
|
|
|
|
| 18 |
from functools import lru_cache
|
| 19 |
from typing import Union
|
| 20 |
|
| 21 |
+
from pathlib import Path
|
| 22 |
import torch
|
| 23 |
import torchaudio
|
| 24 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 2084 |
local_dir = snapshot_download(repo_id)
|
| 2085 |
else:
|
| 2086 |
local_dir = "sherpa-onnx-funasr-nano-int8-2025-12-30"
|
|
|
|
|
|
|
|
|
|
| 2087 |
|
| 2088 |
+
encoder_adaptor = f"{local_dir}/encoder_adaptor.int8.onnx"
|
| 2089 |
+
llm = f"{local_dir}/llm.int8.onnx"
|
| 2090 |
+
embedding = f"{local_dir}/embedding.int8.onnx"
|
| 2091 |
+
tokenizer = f"{local_dir}/Qwen3-0.6B"
|
| 2092 |
+
if not Path(encoder_adaptor).is_file():
|
| 2093 |
+
os.system(
|
| 2094 |
+
f"""
|
| 2095 |
+
mkdir -p {local_dir}
|
| 2096 |
+
cd {local_dir}
|
| 2097 |
+
wget https://huggingface.co/csukuangfj/sherpa-onnx-funasr-nano-int8-2025-12-30/resolve/main/embedding.int8.onnx
|
| 2098 |
+
wget https://huggingface.co/csukuangfj/sherpa-onnx-funasr-nano-int8-2025-12-30/resolve/main/encoder_adaptor.int8.onnx
|
| 2099 |
+
wget https://huggingface.co/csukuangfj/sherpa-onnx-funasr-nano-int8-2025-12-30/resolve/main/llm.int8.onnx
|
| 2100 |
+
mkdir Qwen3-0.6B
|
| 2101 |
+
cd Qwen3-0.6B
|
| 2102 |
+
wget https://huggingface.co/csukuangfj/sherpa-onnx-funasr-nano-int8-2025-12-30/resolve/main/Qwen3-0.6B/merges.txt
|
| 2103 |
+
wget https://huggingface.co/csukuangfj/sherpa-onnx-funasr-nano-int8-2025-12-30/resolve/main/Qwen3-0.6B/tokenizer.json
|
| 2104 |
+
wget https://huggingface.co/csukuangfj/sherpa-onnx-funasr-nano-int8-2025-12-30/resolve/main/Qwen3-0.6B/vocab.json
|
| 2105 |
+
"""
|
| 2106 |
+
)
|
| 2107 |
|
| 2108 |
+
os.system(f"ls -lh {local_dir}")
|
|
|
|
|
|
|
|
|
|
| 2109 |
|
| 2110 |
print("local_dir", local_dir)
|
| 2111 |
print("encoder_adaptor", encoder_adaptor)
|