Skip to content

GitLab

Unverified Commit df15703b authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix doc builder Dockerfile (#17435) 



* Fix doc builder Dockerfile
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent a72f1c9f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
docker/transformers-doc-builder/Dockerfile
View file @ df15703b
...@@ -15,5 +15,6 @@ RUN python3 -m pip install --no-cache-dir torchvision git+https://github.com/fac ...@@ -15,5 +15,6 @@ RUN python3 -m pip install --no-cache-dir torchvision git+https://github.com/fac
RUN python3 -m pip install --no-cache-dir pytorch-quantization --extra-index-url https://pypi.ngc.nvidia.com RUN python3 -m pip install --no-cache-dir pytorch-quantization --extra-index-url https://pypi.ngc.nvidia.com
RUN python3 -m pip install -U "itsdangerous<2.1.0" RUN python3 -m pip install -U "itsdangerous<2.1.0"
RUN doc-builder build transformers transformers/docs/source --build_dir doc-build-dev --notebook_dir notebooks/transformers_doc --clean --version pr_$PR_NUMBER # Test if the image could successfully build the doc. before publishing the image
RUN doc-builder build transformers transformers/docs/source/en --build_dir doc-build-dev --notebook_dir notebooks/transformers_doc --clean
RUN rm -rf doc-build-dev RUN rm -rf doc-build-dev
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment