done (#16340)

c30798ec · Francesco Saverio Zuppichini · GitHub · d49f8d31 · c30798ec
Unverified Commit c30798ec authored Mar 22, 2022 by Francesco Saverio Zuppichini Committed by GitHub Mar 22, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/van/modeling_van.py src/transformers/models/van/modeling_van.py +1 -1

No files found.
--- a/src/transformers/models/van/modeling_van.py
+++ b/src/transformers/models/van/modeling_van.py
@@ -43,7 +43,7 @@ _CONFIG_FOR_DOC = "VanConfig"
 _FEAT_EXTRACTOR_FOR_DOC = "AutoFeatureExtractor"

 # Base docstring
-_CHECKPOINT_FOR_DOC = "van-base"
+_CHECKPOINT_FOR_DOC = "Visual-Attention-Network/van-base"
 _EXPECTED_OUTPUT_SHAPE = [1, 512, 7, 7]

 # Image classification docstring