csukuangfj commited on
Commit
4850190
·
1 Parent(s): 108659c

small fixes

Browse files
Files changed (1) hide show
  1. model.py +21 -8
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
- os.system(f"ls -lh {local_dir}")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2091
 
2092
- encoder_adaptor = f"{local_dir}/encoder_adaptor.int8.onnx"
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)