mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
16 lines
1.1 KiB
Text
16 lines
1.1 KiB
Text
|
FROM python:3.8
|
||
|
|
LABEL maintainer="Hugging Face"
|
||
|
|
|
||
|
|
RUN apt update
|
||
|
|
RUN git clone https://github.com/huggingface/transformers
|
||
|
|
|
||
|
|
RUN python3 -m pip install --no-cache-dir --upgrade pip && python3 -m pip install --no-cache-dir git+https://github.com/huggingface/doc-builder ./transformers[dev,deepspeed]
|
||
|
|
RUN apt-get -y update && apt-get install -y libsndfile1-dev && apt install -y tesseract-ocr
|
||
|
|
|
||
|
|
RUN python3 -m pip install --no-cache-dir torch-scatter -f https://data.pyg.org/whl/torch-$(python -c "from torch import version; print(version.__version__.split('+')[0])")+cpu.html
|
||
|
|
RUN python3 -m pip install --no-cache-dir torchvision git+https://github.com/facebookresearch/detectron2.git pytesseract https://github.com/kpu/kenlm/archive/master.zip
|
||
|
|
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 doc-builder build transformers transformers/docs/source --build_dir doc-build-dev --notebook_dir notebooks/transformers_doc --clean --version pr_$PR_NUMBER
|
||
|
|
RUN rm -rf doc-build-dev
|