Pin revision for LayoutLMForQuestionAnswering and TFLayoutLMForQuesti…

…onAnswering tests (huggingface#18854) * Pin revision for tests * Fixup * Update revision in models * Shorten revisions Co-authored-by: Ankur Goyal <ankur@impira.com>
oneraghavan · Sep 26, 2022 · 7e767f6 · 7e767f6
1 parent ec48ecb
commit 7e767f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/transformers/models/layoutlm/modeling_layoutlm.py b/src/transformers/models/layoutlm/modeling_layoutlm.py
@@ -1285,7 +1285,7 @@ def forward(
         >>> import torch
 
         >>> tokenizer = AutoTokenizer.from_pretrained("impira/layoutlm-document-qa", add_prefix_space=True)
-        >>> model = LayoutLMForQuestionAnswering.from_pretrained("impira/layoutlm-document-qa")
+        >>> model = LayoutLMForQuestionAnswering.from_pretrained("impira/layoutlm-document-qa", revision="1e3ebac")
 
         >>> dataset = load_dataset("nielsr/funsd", split="train")
         >>> example = dataset[0]

diff --git a/src/transformers/models/layoutlm/modeling_tf_layoutlm.py b/src/transformers/models/layoutlm/modeling_tf_layoutlm.py
@@ -1448,7 +1448,7 @@ def call(
         >>> from datasets import load_dataset
 
         >>> tokenizer = AutoTokenizer.from_pretrained("impira/layoutlm-document-qa", add_prefix_space=True)
-        >>> model = TFLayoutLMForQuestionAnswering.from_pretrained("impira/layoutlm-document-qa")
+        >>> model = TFLayoutLMForQuestionAnswering.from_pretrained("impira/layoutlm-document-qa", revision="1e3ebac")
 
         >>> dataset = load_dataset("nielsr/funsd", split="train")
         >>> example = dataset[0]