Fix LayoutLM ONNX test error (huggingface#13710)

Fix LayoutLM ONNX test error
lapisfluvialis · Sep 29, 2021 · 1008324 · 1008324
1 parent 11144a3
commit 1008324
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/docs/source/serialization.rst b/docs/source/serialization.rst
@@ -42,6 +42,7 @@ Ready-made configurations include the following models:
 - BERT
 - DistilBERT
 - GPT-2
+- LayoutLM
 - RoBERTa
 - T5
 - XLM-RoBERTa

diff --git a/src/transformers/models/layoutlm/configuration_layoutlm.py b/src/transformers/models/layoutlm/configuration_layoutlm.py
@@ -183,11 +183,6 @@ def generate_dummy_inputs(
             raise ValueError("Cannot generate dummy inputs without PyTorch installed.")
         import torch
 
-        input_dict["bbox"] = torch.tensor(
-            [
-                [0] * 4,
-                *[box] * seq_length,
-                [self.max_2d_positions] * 4,
-            ]
-        ).tile(batch_size, 1, 1)
+        batch_size, seq_length = input_dict["input_ids"].shape
+        input_dict["bbox"] = torch.tensor([*[box] * seq_length]).tile(batch_size, 1, 1)
         return input_dict