mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

support different versions of rknn

Browse Source
This commit is contained in:
Fangjun Kuang 2025-03-19 11:15:26 +08:00
parent 615e5206e1
commit 0dd26db699
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
.github/workflows/rknn.yml vendored
View File

@ -18,7 +18,7 @@ concurrency:
jobs: jobs:
rknn: rknn:
name: RKNN ${{ matrix.recipe }} name: RKNN ${{ matrix.recipe }} ${{ matrix.rknn_toolkit2_version }}
runs-on: ubuntu-latest runs-on: ubuntu-latest
strategy: strategy:
fail-fast: false fail-fast: false
@ -29,7 +29,10 @@ jobs:
torch-version: ["2.0.0"] torch-version: ["2.0.0"]
torchaudio-version: ["2.0.1"] torchaudio-version: ["2.0.1"]
version: ["20241218"] version: ["20241218"]
recipe: ["librispeech", "wenetspeech", "multi_zh-hans"] # recipe: ["librispeech", "wenetspeech", "multi_zh-hans"]
recipe: ["librispeech"]
rknn_toolkit2_version: ["2.2.0", "2.1.0"]
steps: steps:
- uses: actions/checkout@v4 - uses: actions/checkout@v4
@ -55,9 +58,17 @@ jobs:
python3 -m torch.utils.collect_env python3 -m torch.utils.collect_env
python3 -m k2.version python3 -m k2.version
pip list pip list
export rknn_toolkit2_version=${{ matrix.rknn_toolkit2_version }}
if [[ $rknn_toolkit2_version == "2.1.0" ]]; then
# for the folder pruned_transducer_stateless7_streaming
curl -SL -O https://huggingface.co/csukuangfj/rknn-toolkit2/resolve/main/rknn_toolkit2-2.1.0%2B708089d1-cp310-cp310-linux_x86_64.whl
else
# for the folder zipformer/
curl -SL -O https://huggingface.co/csukuangfj/rknn-toolkit2/resolve/main/rknn_toolkit2-2.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
fi
# Install rknn # Install rknn
curl -SL -O https://huggingface.co/csukuangfj/rknn-toolkit2/resolve/main/rknn_toolkit2-2.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
pip install ./*.whl "numpy<=1.26.4" pip install ./*.whl "numpy<=1.26.4"
pip list | grep rknn pip list | grep rknn
echo "---" echo "---"
@ -69,13 +80,13 @@ jobs:
- uses: actions/upload-artifact@v4 - uses: actions/upload-artifact@v4
with: with:
name: log-${{ matrix.recipe }} name: log-${{ matrix.recipe }}-${{ matrix.rknn_toolkit2_version }}
path: ./log-*.txt path: ./log-*.txt
- name: Display results - name: Display results
shell: bash shell: bash
run: | run: |
ls -lh *rk*.tar.bz2 ls -lh *rk*.tar.bz2 || true
- name: Release to GitHub - name: Release to GitHub
uses: svenstaro/upload-release-action@v2 uses: svenstaro/upload-release-action@v2
@ -111,7 +122,7 @@ jobs:
git merge -m "merge remote" --ff origin main git merge -m "merge remote" --ff origin main
dst=streaming-asr dst=streaming-asr
mkdir -p $dst mkdir -p $dst
cp ../*rk*.tar.bz2 $dst/ cp ../*rk*.tar.bz2 $dst/ || true
ls -lh $dst ls -lh $dst
git add . git add .