Skip to content

GitLab

"examples/pytorch/text-classification/run_classification.py" did not exist on "9129fd0377e4d46cb2d0ea28dc1eb91a15f65b77"
Unverified Commit b1706f69 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Install video dependency for pipeline CI (#20777) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent fca66abe
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docker/transformers-pytorch-gpu/Dockerfile
View file @ b1706f69
...@@ -9,7 +9,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip ...@@ -9,7 +9,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip
ARG REF=main ARG REF=main
RUN git clone https://github.com/huggingface/transformers && cd transformers && git checkout $REF RUN git clone https://github.com/huggingface/transformers && cd transformers && git checkout $REF
RUN python3 -m pip install --no-cache-dir -e ./transformers[dev-torch,testing] RUN python3 -m pip install --no-cache-dir -e ./transformers[dev-torch,testing,video]
# If set to nothing, will install the latest version # If set to nothing, will install the latest version
ARG PYTORCH='1.13.0' ARG PYTORCH='1.13.0'
......
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