NiniAndy commited on
Commit
2f44b6e
·
verified ·
1 Parent(s): 8315d00

Pin validated inference dependencies and refresh setup guides

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  2. README.zh.md +2 -2
  3. requirements.txt +10 -8
README.md CHANGED
@@ -38,7 +38,7 @@ English · [简体中文](https://huggingface.co/inclusionAI/Ming-3.0-asr-tiny/b
38
 
39
  ### Download and install
40
 
41
- Prepare compatible CUDA builds of PyTorch and torchaudio, then download the complete checkpoint and install its remaining dependencies:
42
 
43
  ```bash
44
  python -m pip install "huggingface_hub[cli]"
@@ -56,7 +56,7 @@ python examples/transcribe.py \
56
  --audio-chunk-frames 6000
57
  ```
58
 
59
- Add `--language-hint zh` or `--language-hint en` when the language is known. The example uses CUDA, BF16, SDPA, and greedy decoding; it loads the bundled model code with `trust_remote_code=True`. The reference environment used Transformers 5.12.1, fla-core 0.5.2, and PyTorch 2.13.0+cu130.
60
 
61
  ### PyTorch / Transformers Python API
62
 
 
38
 
39
  ### Download and install
40
 
41
+ Use Python 3.12 on a Linux machine with a compatible CUDA GPU. Download the complete checkpoint and install the tested dependencies, including PyTorch and torchaudio:
42
 
43
  ```bash
44
  python -m pip install "huggingface_hub[cli]"
 
56
  --audio-chunk-frames 6000
57
  ```
58
 
59
+ Add `--language-hint zh` or `--language-hint en` when the language is known. The example uses CUDA, BF16, SDPA, and greedy decoding; it loads the bundled model code with `trust_remote_code=True`. The tested environment used Transformers 5.12.1, fla-core 0.5.2, and PyTorch 2.11.0+cu130. For another CUDA version, first install matching PyTorch and torchaudio 2.11.0 wheels using the [official PyTorch instructions](https://pytorch.org/get-started/previous-versions/), then run `pip install -r requirements.txt`.
60
 
61
  ### PyTorch / Transformers Python API
62
 
README.zh.md CHANGED
@@ -25,7 +25,7 @@
25
 
26
  ### 下载与安装
27
 
28
- 先准备兼容 CUDA 的 PyTorch 和 torchaudio,再下载完整模型并安装其余依赖:
29
 
30
  ```bash
31
  python -m pip install "huggingface_hub[cli]"
@@ -43,7 +43,7 @@ python examples/transcribe.py \
43
  --audio-chunk-frames 6000
44
  ```
45
 
46
- 已知语种时可加 `--language-hint zh` 或 `--language-hint en`。示例使用 CUDA、BF16、SDPA 和 greedy decoding,并通过 `trust_remote_code=True` 加载仓库内的自定义实现。参考验证环境为 Transformers 5.12.1、fla-core 0.5.2 和 PyTorch 2.13.0+cu130。
47
 
48
  ### PyTorch / Transformers Python 示例
49
 
 
25
 
26
  ### 下载与安装
27
 
28
+ 使用 Python 3.12 和兼容 CUDA 的 Linux GPU。下载完整模型后,按已验证的依赖版本安装,包括 PyTorch 与 torchaudio:
29
 
30
  ```bash
31
  python -m pip install "huggingface_hub[cli]"
 
43
  --audio-chunk-frames 6000
44
  ```
45
 
46
+ 已知语种时可加 `--language-hint zh` 或 `--language-hint en`。示例使用 CUDA、BF16、SDPA 和 greedy decoding,并通过 `trust_remote_code=True` 加载仓库内的自定义实现。已验证环境为 Transformers 5.12.1、fla-core 0.5.2 和 PyTorch 2.11.0+cu130。若需其他 CUDA 版本,先按 [PyTorch 官方说明](https://pytorch.org/get-started/previous-versions/)安装匹配的 PyTorch 与 torchaudio 2.11.0 wheel,再运行 `pip install -r requirements.txt`。
47
 
48
  ### PyTorch / Transformers Python 示例
49
 
requirements.txt CHANGED
@@ -1,9 +1,11 @@
1
- # Runtime interface validated with Transformers 5.12.1 and fla-core 0.5.2.
2
- # Install mutually compatible CUDA builds of torch and torchaudio first.
3
- transformers>=5.12.1,<5.13
 
 
4
  fla-core==0.5.2
5
- accelerate
6
- einops
7
- safetensors
8
- numpy
9
- soundfile
 
1
+ # Validated on Python 3.12, Linux, CUDA 13.0 (NVIDIA L20Y).
2
+ # For another CUDA version, install matching PyTorch 2.11.0 wheels first.
3
+ torch==2.11.0
4
+ torchaudio==2.11.0
5
+ transformers==5.12.1
6
  fla-core==0.5.2
7
+ accelerate==1.15.0
8
+ einops==0.8.2
9
+ safetensors==0.8.0
10
+ numpy==2.5.4
11
+ soundfile==0.14.0