SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Fix doc example] Fix 2 PyTorch Vilt docstring examples (#16076)

Browse Source 
* fix 2 pytorch vilt docstring examples

* add vilt to doctest list file

* remove device

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-03-15 13:35:02 +01:00
committed by GitHub
parent bcaf566038
commit e5bc438cc8
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
utils/documentation_tests.txt
View File

@@ -18,6 +18,7 @@ src/transformers/models/swin/modeling_swin.py
src/transformers/models/convnext/modeling_convnext.py
src/transformers/models/poolformer/modeling_poolformer.py
src/transformers/models/vit_mae/modeling_vit_mae.py
src/transformers/models/vilt/modeling_vilt.py
src/transformers/models/van/modeling_van.py
src/transformers/models/segformer/modeling_segformer.py
src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py