Build doc

huggingface · Dec 19, 2024 · b447d1a · b447d1a
1 parent 9309d77
commit b447d1a
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 41 deletions.
diff --git a/.github/workflows/build_pr_documentation.yml b/.github/workflows/build_pr_documentation.yml
@@ -70,45 +70,45 @@ jobs:
           pip install black
           cd ..
 
-      - name: Make Habana documentation
-        run: |
-          sudo docker system prune -a -f
-          cd optimum-habana
-          make doc BUILD_DIR=habana-doc-build VERSION=pr_$PR_NUMBER
-          sudo mv habana-doc-build ../optimum
-          cd ..
-
-      - name: Make Intel documentation
-        run: |
-          sudo docker system prune -a -f
-          cd optimum-intel
-          make doc BUILD_DIR=intel-doc-build VERSION=pr_$PR_NUMBER
-          sudo mv intel-doc-build ../optimum
-          cd ..
-
-      # TODO: enable Furiosa doc build in PRs once archive.furiosa.ai is public
-      - name: Make Furiosa documentation
-        run: |
-          echo "For PRs we don't build Furiosa doc"
-
-      - name: Make AMD documentation
-        run: |
-          sudo docker system prune -a -f
-          cd optimum-amd
-          make doc BUILD_DIR=amd-doc-build VERSION=pr_$PR_NUMBER
-          sudo mv amd-doc-build ../optimum
-          cd ..
-
-      - name: Make TPU documentation
-        run: |
-          sudo docker system prune -a -f
-          source venv-doc/bin/activate
-          cd optimum-tpu
-          pip install -U pip
-          pip install . -f https://storage.googleapis.com/libtpu-releases/index.html
-          doc-builder build optimum.tpu docs/source/ --build_dir tpu-doc-build --version pr_$PR_NUMBER --version_tag_suffix "" --html --clean
-          mv tpu-doc-build ../optimum
-          cd ..
+          # - name: Make Habana documentation
+          #   run: |
+          #     sudo docker system prune -a -f
+          #     cd optimum-habana
+          #     make doc BUILD_DIR=habana-doc-build VERSION=pr_$PR_NUMBER
+          #     sudo mv habana-doc-build ../optimum
+          #     cd ..
+
+          # - name: Make Intel documentation
+          #   run: |
+          #     sudo docker system prune -a -f
+          #     cd optimum-intel
+          #     make doc BUILD_DIR=intel-doc-build VERSION=pr_$PR_NUMBER
+          #     sudo mv intel-doc-build ../optimum
+          #     cd ..
+
+          # # TODO: enable Furiosa doc build in PRs once archive.furiosa.ai is public
+          # - name: Make Furiosa documentation
+          #   run: |
+          #     echo "For PRs we don't build Furiosa doc"
+
+          # - name: Make AMD documentation
+          #   run: |
+          #     sudo docker system prune -a -f
+          #     cd optimum-amd
+          #     make doc BUILD_DIR=amd-doc-build VERSION=pr_$PR_NUMBER
+          #     sudo mv amd-doc-build ../optimum
+          #     cd ..
+
+          # - name: Make TPU documentation
+          #   run: |
+          #     sudo docker system prune -a -f
+          #     source venv-doc/bin/activate
+          #     cd optimum-tpu
+          #     pip install -U pip
+          #     pip install . -f https://storage.googleapis.com/libtpu-releases/index.html
+          #     doc-builder build optimum.tpu docs/source/ --build_dir tpu-doc-build --version pr_$PR_NUMBER --version_tag_suffix "" --html --clean
+          #     mv tpu-doc-build ../optimum
+          #     cd ..
 
       - name: Make Optimum documentation
         run: |

diff --git a/docs/Dockerfile b/docs/Dockerfile
@@ -8,5 +8,4 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip
 RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/doc-builder.git
 
 RUN git clone $clone_url && cd optimum && git checkout $commit_sha
-RUN python3 -m pip install --no-cache-dir ./optimum[onnxruntime,benchmark,quality,exporters-tf,doc-build,diffusers]
-RUN python3 -m pip install ./optimum[exporters-executorch]
+RUN python3 -m pip install --no-cache-dir ./optimum[onnxruntime,benchmark,quality,exporters-tf,exporters-executorch,doc-build,diffusers] --index-url https://download.pytorch.org/whl/cpu