Skip to content

Commit 82bb682

Browse files
NielsRoggeNiels Rogge
andauthored
[VideoMAE] Add model to doc tests (#18523)
* Add videomae to doc tests * Add pip install decord Co-authored-by: Niels Rogge <[email protected]>
1 parent 3632531 commit 82bb682

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

docker/transformers-all-latest-gpu/Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@ RUN python3 -m pip install -U "itsdangerous<2.1.0"
4545

4646
RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/accelerate@main#egg=accelerate
4747

48+
RUN python3 -m pip install --no-cache-dir decord
49+
4850
# When installing in editable mode, `transformers` is not recognized as a package.
4951
# this line must be added in order for python to be aware of transformers.
5052
RUN cd transformers && python3 setup.py develop

utils/documentation_tests.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@ src/transformers/models/trocr/modeling_trocr.py
7070
src/transformers/models/unispeech/modeling_unispeech.py
7171
src/transformers/models/unispeech_sat/modeling_unispeech_sat.py
7272
src/transformers/models/van/modeling_van.py
73+
src/transformers/models/videomae/modeling_videomae.py
7374
src/transformers/models/vilt/modeling_vilt.py
7475
src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py
7576
src/transformers/models/vit/modeling_vit.py

0 commit comments

Comments
 (0)