refactor: leverage ExecutorTorch's CustomKVCache for per-layer cache … #38
This file contains hidden or 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 documentation | |
| on: | |
| push: | |
| branches: | |
| - main | |
| tags: | |
| - 'v[0-9]+.[0-9]+.[0-9]+' | |
| workflow_dispatch: | |
| jobs: | |
| build_documentation: | |
| runs-on: ubuntu-22.04 | |
| env: | |
| COMMIT_SHA: ${{ github.event.pull_request.head.sha }} | |
| PR_NUMBER: ${{ github.event.number }} | |
| EVENT_CONTEXT: ${{ toJSON(github.event) }} | |
| PR_CLONE_URL: ${{ github.event.pull_request.head.repo.clone_url }} | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-node@v4 | |
| with: | |
| node-version: '18' | |
| cache-dependency-path: "kit/package-lock.json" | |
| - name: Set up Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.11' | |
| - name: Set environment variables | |
| run: | | |
| cd optimum | |
| version=`echo "$(grep '^__version__ =' executorch/version.py | cut -d '=' -f 2- | xargs)"` | |
| if [[ $version == *.dev0 ]] | |
| then | |
| echo "VERSION=main" >> $GITHUB_ENV | |
| else | |
| echo "VERSION=v$version" >> $GITHUB_ENV | |
| fi | |
| cd .. | |
| - name: Setup environment | |
| run: | | |
| python -m pip install --upgrade pip | |
| python -m ensurepip --upgrade | |
| python -m pip install --upgrade setuptools | |
| python -m pip install git+https://github.com/huggingface/doc-builder | |
| python -m pip install .[quality] | |
| - name: Make documentation | |
| shell: bash | |
| run: | | |
| doc-builder build optimum.executorch docs/source/ \ | |
| --repo_name optimum-executorch \ | |
| --build_dir executorch-doc-build/ \ | |
| --version ${{ env.VERSION }} \ | |
| --version_tag_suffix "" \ | |
| --html \ | |
| --clean \ | |
| - name: Push documentation | |
| run: | | |
| sudo chmod -R ugo+rwx executorch-doc-build | |
| cd executorch-doc-build | |
| sudo mv optimum.executorch optimum-executorch | |
| doc-builder push optimum-executorch --doc_build_repo_id "hf-doc-build/doc-build" --token "${{ secrets.HF_DOC_BUILD_PUSH }}" --commit_msg "Updated with commit $COMMIT_SHA See: https://github.com/huggingface/optimum-executorch/commit/$COMMIT_SHA" --n_retries 5 |