Export Pyannote models to torchscript (#694) #4
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: build-wheels-cpu-ubuntu | |
| on: | |
| push: | |
| tags: | |
| - '*' | |
| workflow_dispatch: | |
| concurrency: | |
| group: build-wheels-cpu-ubuntu-${{ github.ref }} | |
| cancel-in-progress: true | |
| jobs: | |
| generate_build_matrix: | |
| # see https://github.com/pytorch/pytorch/pull/50633 | |
| runs-on: ubuntu-latest | |
| outputs: | |
| matrix: ${{ steps.set-matrix.outputs.matrix }} | |
| steps: | |
| - uses: actions/checkout@v2 | |
| with: | |
| fetch-depth: 0 | |
| - name: Generating build matrix | |
| id: set-matrix | |
| run: | | |
| # outputting for debugging purposes | |
| python ./scripts/github_actions/generate_build_matrix.py | |
| MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py) | |
| echo "::set-output name=matrix::${MATRIX}" | |
| build-manylinux-wheels: | |
| needs: generate_build_matrix | |
| name: ${{ matrix.torch }} ${{ matrix.python-version }} | |
| runs-on: ubuntu-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| ${{ fromJson(needs.generate_build_matrix.outputs.matrix) }} | |
| steps: | |
| - uses: actions/checkout@v2 | |
| with: | |
| fetch-depth: 0 | |
| - name: Setup Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v2 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Display Python version | |
| run: python -c "import sys; print(sys.version)" | |
| - name: Install patchelf | |
| shell: bash | |
| run: | | |
| sudo apt-get update -q | |
| sudo apt-get install -q -y patchelf | |
| patchelf --help | |
| # see https://github.com/pytorch/test-infra/blob/9e3d392690719fac85bad0c9b67f530e48375ca1/tools/scripts/generate_binary_build_matrix.py | |
| # https://github.com/pytorch/builder/tree/main/manywheel | |
| # https://github.com/pytorch/builder/pull/476 | |
| # https://github.com/k2-fsa/k2/issues/733 | |
| # https://github.com/pytorch/pytorch/pull/50633 (generate build matrix) | |
| - name: Run the build process with Docker | |
| uses: addnab/docker-run-action@v3 | |
| with: | |
| image: ${{ matrix.image }} | |
| options: -v ${{ github.workspace }}:/var/www -e PYTHON_VERSION=${{ matrix.python-version }} -e TORCH_VERSION=${{ matrix.torch }} | |
| run: | | |
| echo "pwd: $PWD" | |
| uname -a | |
| id | |
| cat /etc/*release | |
| gcc --version | |
| python3 --version | |
| which python3 | |
| /var/www/scripts/github_actions/build-ubuntu-cpu.sh | |
| - name: Patch wheels | |
| shell: bash | |
| run: | | |
| ls -lh | |
| ls -lh ./wheels/ | |
| sudo ./scripts/github_actions/patch_wheel.py --in-dir ./wheels --out-dir ./wheelhouse | |
| ls -lh ./wheelhouse/ | |
| - name: Upload Wheel | |
| uses: actions/upload-artifact@v2 | |
| with: | |
| name: torch-${{ matrix.torch }}-python-${{ matrix.python-version }}-ubuntu-latest-cpu | |
| path: wheelhouse/*.whl | |
| # https://huggingface.co/docs/hub/spaces-github-actions | |
| - name: Publish to huggingface | |
| if: github.repository_owner == 'csukuangfj' || github.repository_owner == 'k2-fsa' | |
| env: | |
| HF_TOKEN: ${{ secrets.HF_TOKEN }} | |
| uses: nick-fields/retry@v2 | |
| with: | |
| max_attempts: 20 | |
| timeout_seconds: 200 | |
| shell: bash | |
| command: | | |
| git config --global user.email "[email protected]" | |
| git config --global user.name "Fangjun Kuang" | |
| rm -rf huggingface | |
| export GIT_LFS_SKIP_SMUDGE=1 | |
| git clone https://huggingface.co/csukuangfj/sherpa huggingface | |
| cd huggingface | |
| git pull | |
| mkdir -p ubuntu-cpu | |
| cp -v ../wheelhouse/*.whl ./ubuntu-cpu | |
| git status | |
| git lfs track "*.whl" | |
| git add . | |
| git commit -m "upload ubuntu-cpu wheel for torch ${{ matrix.torch }} python ${{ matrix.python-version }}" | |
| git push https://csukuangfj:[email protected]/csukuangfj/sherpa main |