Skip to content

Enable INC for llava models and change softmax to use torch.nn.functional.softmax as its supported module by INC #3559

Enable INC for llava models and change softmax to use torch.nn.functional.softmax as its supported module by INC

Enable INC for llava models and change softmax to use torch.nn.functional.softmax as its supported module by INC #3559

name: Build PR documentation
on:
pull_request:
branches: [ main ]
concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true
jobs:
build_documentation:
runs-on: ubuntu-20.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@v2
with:
repository: 'huggingface/doc-builder'
path: doc-builder
- uses: actions/checkout@v2
with:
repository: 'huggingface/optimum-habana'
path: optimum-habana
- name: Setup environment
run: |
pip uninstall -y doc-builder
cd doc-builder
git pull origin main
pip install .
cd ..
- name: Make documentation
run: |
cd optimum-habana
make doc BUILD_DIR=habana-doc-build VERSION=pr_$PR_NUMBER COMMIT_SHA_SUBPACKAGE=$COMMIT_SHA CLONE_URL=$PR_CLONE_URL
cd ..
- name: Save commit_sha & pr_number
run: |
cd optimum-habana
sudo chmod -R ugo+rwx habana-doc-build
cd habana-doc-build
sudo mv optimum.habana optimum-habana
echo ${{ env.COMMIT_SHA }} > ./commit_sha
echo ${{ env.PR_NUMBER }} > ./pr_number
- uses: actions/upload-artifact@v3
with:
name: doc-build-artifact
path: optimum-habana/habana-doc-build/