mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-10 02:22:17 +00:00
77 lines
1.6 KiB
YAML
77 lines
1.6 KiB
YAML
name: rknn
|
|
|
|
on:
|
|
push:
|
|
branches:
|
|
- master
|
|
- ci-rknn
|
|
|
|
pull_request:
|
|
branches:
|
|
- master
|
|
|
|
workflow_dispatch:
|
|
|
|
concurrency:
|
|
group: rknn-${{ github.ref }}
|
|
cancel-in-progress: true
|
|
|
|
jobs:
|
|
rknn:
|
|
runs-on: ubuntu-latest
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
os: [ubuntu-latest]
|
|
python-version: ["3.10"]
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
with:
|
|
fetch-depth: 0
|
|
|
|
- name: Setup Python
|
|
uses: actions/setup-python@v5
|
|
with:
|
|
python-version: ${{ matrix.python-version }}
|
|
|
|
- name: Install rknn
|
|
shell: bash
|
|
run: |
|
|
ls
|
|
curl -SL -O https://huggingface.co/csukuangfj/rknn-toolkit2/resolve/main/rknn_toolkit2-1.6.2b6%2B2a4e0d97-cp310-cp310-linux_x86_64.whl
|
|
pip install ./*.whl "numpy<=1.26.4"
|
|
pip list | grep rknn
|
|
echo "---"
|
|
pip list
|
|
echo "---"
|
|
|
|
- name: Run
|
|
shell: bash
|
|
run: |
|
|
ls
|
|
python3 -c "import sys; from rknn.api import RKNN; print(dir(RKNN))"
|
|
|
|
git clone --depth 1 https://github.com/airockchip/rknn_model_zoo/
|
|
cd rknn_model_zoo/examples/zipformer
|
|
cd model
|
|
ls -lh
|
|
chmod +x ./download_model.sh
|
|
./download_model.sh
|
|
ls -lh
|
|
cd ..
|
|
|
|
cd python
|
|
python3 ./convert.py ../model/encoder-epoch-99-avg-1.onnx rk3588
|
|
|
|
ls -lh ../model
|
|
|
|
python3 convert.py ../model/decoder-epoch-99-avg-1.onnx rk3588
|
|
|
|
ls -lh ../model
|
|
|
|
python3 convert.py ../model/joiner-epoch-99-avg-1.onnx rk3588
|
|
|
|
ls -lh ../model
|
|
|
|
|